/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 vote1.2K views1 answer
How to print cow on python
Python
89 votes91 views6 answers
Int to string python
Python
0 vote140 views1 answer
Somalia embassy in bangladesh
Python
0 vote124 views1 answer
F readlines python not working
Python
2 votes81 views1 answer
Check if user has manage messages discord.py
Python
1 vote135 views1 answer
Django boilerplate command
Python
57 votes136 views6 answers
Timedelta python
Python
6 votes153 views3 answers
Django models filter
Python
0 vote129 views1 answer
Tqdm every new line
Python
4 votes152 views1 answer
Any python
Python