/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

-1 vote173 views2 answers
Model deployment with flask
Python Flask
8 votes189 views1 answer
Numpy where
Python
2 votes247 views2 answers
Python check string for negative number
Python
0 vote212 views1 answer
Unable to import flask pylint
Python
6 votes279 views1 answer
Print url selenium python
Python
5 votes266 views3 answers
Sys module in python
Python
15 votes201 views3 answers
Divide list into equal parts python
Python
1 vote191 views2 answers
How to append a tuple to a list
Python
4 votes144 views1 answer
Pandas display one row
Python
2 votes181 views1 answer
Django models distinct
Python