/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

3 votes197 views2 answers
List of dictionary values
Python
1 vote100 views3 answers
Mail mailer
Python
4 votes154 views2 answers
ImportError: No module named django.core.wsgi
Python
0 vote158 views2 answers
Headers cors
Python
0 vote160 views1 answer
Plt add y gridlines
Python
0 vote162 views1 answer
Return the biggest even fro a list python
Python
0 vote144 views1 answer
Python file hashlib
Python
0 vote149 views3 answers
Datetime columns only extract date pandas
Python
3 votes150 views3 answers
Quadratic equation
Python
5 votes149 views1 answer
Trimming spaces in string python
Python