pytorch读取数据集问题
当用自己的图数据跑别人的图神经网络代码时出现的问题,数据集的数据不同,结果出错。
错误:
# size [batch_size, cHeads, centrs, graphsize]
dist = torch.sum(torch.abs(points - batch_centroids_broad) ** 2, 4)
RuntimeError: The size of tensor a (450) must match the size of tensor b (5) at non-singleton..
原创
2021-11-25 10:01:02 ·
1603 阅读 ·
0 评论