/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

35 votes152 views2 answers
Convert into date python
Python
0 vote153 views1 answer
How to slice few rows in pandas
Python
0 vote148 views1 answer
Create new column with mask pandas
Python
2 votes173 views6 answers
Convert ui to py
Python
0 vote145 views1 answer
Python format float as currency
Python
0 vote155 views1 answer
Verilog import module
Python
3 votes162 views2 answers
Change django administration text
Python
3 votes188 views2 answers
Python script that executes at time
Python
1 vote204 views1 answer
Add background image in django uploaded file
Python Django
1 vote153 views3 answers
Flatmap python
Python