“Convertir Torch.nn.Embedding capa a una matriz Numpy” Código de respuesta

Convertir Torch.nn.Embedding capa a una matriz Numpy

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)