/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 vote140 views1 answer
Python Tkinter timer animation
Python
1 vote134 views1 answer
Who wrote permission to dance
Python
1 vote131 views2 answers
Create requirements.txt conda
Python
0 vote175 views1 answer
Print map object python
Python
0 vote126 views2 answers
Loop through list of tuples python
Python
0 vote157 views1 answer
Django.core.exceptions.ImproperlyConfigured: Field name is not valid for model
Python Django
0 vote133 views1 answer
Créer un dict python avec une liste
Python
16 votes129 views3 answers
Python split range equally
Python
1 vote138 views1 answer
Pytz
Python
84 votes342 views2 answers
Python with statement file does not exist exception
Python