Learn practical skills, build real-world projects, and advance your career
Created 5 years ago
import torch
import numpy as nptorch.Tensor([-1])tensor([-1.])# 关于torch.nn.Embedding
entity_number = 5
relation_number = 3
dim = 4
ent_embeddings = torch.nn.Embedding(entity_number,dim)
rel_embeddings = torch.nn.Embedding(relation_number,dim)# torch.nn.Embeding.weight & torch.nn.Embeding.weight.data
print("ent_embeddings.weight:",ent_embeddings.weight)
print("ent_embeddings.weight.data:",ent_embeddings.weight.data)ent_embeddings.weight: Parameter containing:
tensor([[-2.5187e-01, 5.8727e-01, 1.2354e+00, 7.1489e-01],
[-5.6152e-01, -5.5336e-01, -1.0236e+00, -4.4449e-01],
[-7.8346e-04, -5.8843e-01, -1.4657e+00, -1.2068e+00],
[ 5.5088e-01, 1.7039e+00, 2.1763e-01, -8.9384e-01],
[ 1.3281e+00, -3.2343e-01, -1.0004e+00, -6.3203e-01]],
requires_grad=True)
ent_embeddings.weight.data: tensor([[-2.5187e-01, 5.8727e-01, 1.2354e+00, 7.1489e-01],
[-5.6152e-01, -5.5336e-01, -1.0236e+00, -4.4449e-01],
[-7.8346e-04, -5.8843e-01, -1.4657e+00, -1.2068e+00],
[ 5.5088e-01, 1.7039e+00, 2.1763e-01, -8.9384e-01],
[ 1.3281e+00, -3.2343e-01, -1.0004e+00, -6.3203e-01]])
batch_h = np.zeros(2,dtype=np.int64)
batch_h[0] = 1
print(batch_h)[1 0]