/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 vote243 views1 answer
Windows how to store filepath as variabley python
Python
0 vote223 views1 answer
Request streaming python
Python
0 vote169 views1 answer
Combine picture and audio python
Python
0 vote421 views1 answer
Django on delete set default
Python Django
0 vote171 views1 answer
Numpy euclidean distance matrix broadcasting
Python
0 vote207 views1 answer
Remove particular row number in pandas
Python
0 vote203 views1 answer
Python developer salary
Python
0 vote253 views1 answer
Arabic text recognition from pdf using python
Python
1 vote168 views1 answer
Display max number of columns pandas
Python
0 vote185 views3 answers
Brew PIP
Python