/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 vote229 views1 answer
Db_index django
Python Django
19 votes186 views2 answers
Python datetime string
Python
9 votes193 views2 answers
Splitting a number into digits python
Python
0 vote194 views1 answer
Convert python float list to 2 digit
Python
6 votes191 views4 answers
Uninstall python 2.7 in ubuntu
Python
16 votes193 views5 answers
Convert pandas dataframe/ table to python dictionary
Python
2 votes184 views1 answer
Download pdf from link using python
Python
0 vote188 views1 answer
Search for a word in pdf using python
Python
9 votes225 views1 answer
How to send get request python
Python
0 vote201 views1 answer
Installing intel python
Python