/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 vote127 views1 answer
Pandas read from txt separtion
Python
1 vote106 views2 answers
How to get pc name with python
Python
0 vote111 views1 answer
Access data in one python function from another
Python
0 vote108 views1 answer
Counting combinations python
Python
0 vote166 views1 answer
How to deal with SettingWithCopyWarning in Pandas
Python
0 vote104 views1 answer
How to visualize decision tree in python
Python
0 vote111 views1 answer
Python pprint on file
Python
0 vote137 views1 answer
Python open all files of type csv
Python
0 vote66 views2 answers
Access matrix value opencv
Python
1 vote123 views1 answer
Python define 2d table
Python