错误
RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'mat2'
异常代码:
def build_adjacency(self, g):
。。。。。。。。。(部分代码省略)
adjacency = np.zeros([len(adj_dict), len(adj_dict)])
for key, value in adj_dict.items():
for item in value:
adjacency[key][item] = 1
return torch.tensor(adjacency)
adjacency = self.build_adjacency(g)
# 此行报错
h = torch.mm(adjacency, h_feature)
解决办法
return torch.tensor(adjacency,dtype=torch.float32)
方法返回值中加上----dtype=torch.float32–即可