/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 vote290 views3 answers
How to get the code of a website in python
Python
0 vote242 views1 answer
Django pull from Google Sheets
Python Django
3 votes223 views1 answer
Get python directiory
Python
0 vote248 views1 answer
Split pandas dataframe in two
Python
2 votes231 views3 answers
Python how to get directory of script
Python
1 vote250 views1 answer
Get random number positive or negative python
Python
21 votes260 views6 answers
How to concatenate two lists in python
Python
4 votes235 views3 answers
Distance formula in python
Python
0 vote239 views1 answer
Python Multiline docstring example
Python
9 votes254 views1 answer
Bee movie script
Python