/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 vote405 views1 answer
Check pip installed packages inside virtualenv
Python
2 votes367 views5 answers
Swap keys and values in dictionary python
Python
1 vote339 views1 answer
Python regex get string before character
Python
0 vote346 views1 answer
Set change order python
Python
10 votes364 views2 answers
Python get numbers after decimal point
Python
1 vote357 views1 answer
Python program for swapping position of two numbers
Python
0 vote356 views1 answer
ValueError: expected sparse matrix with integer values, found float values
Python
1 vote347 views1 answer
Split a text file into multiple paragraphs python
Python
0 vote379 views1 answer
Tkmessagebox not found
Python
0 vote342 views1 answer
Latex maths to python parser
Python