/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 vote123 views1 answer
How to get total number of rows in listbox tkinter
Python
0 vote134 views1 answer
Jupyter notebook don't show plot
Python
0 vote191 views1 answer
Python print stderr
Python
0 vote129 views1 answer
Convert 2 level nested list to one level list in python
Python
1 vote126 views1 answer
Text to sound python
Python
0 vote137 views1 answer
Color plt
Python
0 vote160 views1 answer
Drop nulll python
Python
3 votes127 views2 answers
Wikipedia python
Python
0 vote146 views1 answer
How to print 's in python
Python
0 vote131 views1 answer
Accumulator programming python
Python