Answers for "Convert torch.nn.Embedding layer to numpy array"

0

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)
Posted by: Guest on September-28-2020

Code answers related to "Convert torch.nn.Embedding layer to numpy array"

Python Answers by Framework

Browse Popular Code Answers by Language