/Python/Convert torch.nn.Embedding layer to numpy array

Convert torch.nn.Embedding layer to numpy array

bert_embeddings = bert_model.get_input_embeddings()

# Convert bert embeddings from a torch.nn.Module type to a numpy array
bert_embedding_numpy = np.array(bert_embeddings.weight.data)

Source: discuss.pytorch.org

Similar codes

0 vote91 views1 answer
Join mulitple dataframe pandas index
Python
8 votes91 views3 answers
Linear search python
Python
0 vote99 views1 answer
Django redirect to external url
Python
0 vote113 views1 answer
How to use idl in python
Python
0 vote113 views1 answer
Sublime python build system
Python
1 vote106 views1 answer
Cursor python
Python
0 vote98 views1 answer
Linux desktop files location python
Python
0 vote105 views1 answer
Turn off subplot
Python
3 votes92 views2 answers
Read specific columns from csv in python pandas
Python
0 vote98 views1 answer
How do you see if a data type is an integer python
Python