/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

3 votes242 views4 answers
Fraction in python
Python
1 vote251 views1 answer
How to ask someone for their name in python
Python
1 vote220 views2 answers
Rank function in pandas
Python
5 votes254 views6 answers
Best python ide
Python
0 vote261 views1 answer
How to change the height of an input in python tkinter
Python
7 votes225 views1 answer
Sdjflk
Python
4 votes249 views2 answers
Python replace by dictionary
Python
0 vote265 views1 answer
%%writefile jupyter
Python
16 votes253 views6 answers
How to take integer input in python
Python
29 votes218 views6 answers
JUPYTER INSTALL
Python