Pytorch几何中的edge_weight和edge_attr有什么区别

Question

我想在 Pytorch Geometric 中处理加权无向图。节点特征是50维的。我发现这可以通过类x的属性来处理torch_geometric.data.data。边的权重是标量值。我们发现edge_attr和edge_weight是处理边缘的属性。

我想我应该使用edge_weight，这是正确的吗？

edge_attr另外，和 之间有什么区别edge_weight？

我的英语不太好，所以对此我深表歉意。我希望我能得到一个好的答案。

谢谢。

Answer 1

edge_weight和之间的区别edge_attr在于，它edge_weight始终是一维的（每条边一个值）并且edge_attribute可以是多维的。您可以查看备忘单以了解模型的支持情况。