graph - Pytorch Geometric中的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可以是多维的。您可以查看备忘单以获取模型的支持。