DGL 版本:0.6
官方文件说
- DGLGraph.pull
沿指定边类型从指定节点的前任中提取消息,聚合它们以更新节点特征。
它可以使用节点列表或单个节点作为输入来调用,但是它们的行为似乎并不相同。在我的代码中,调用
dgl_graph.prop_nodes(0)
dgl_graph.prop_nodes(1)
dgl_graph.prop_nodes(2)
顺序很好,但打电话
tree.dgl_graph.pull([0,5,6])
给出以下错误
DGLError: 无法使用方案 Scheme(shape=(1, 128), dtype=torch.float32) 的功能更新方案 Scheme(shape=(3, 128), dtype=torch.float32) 的列。
是否有任何文档可以了解使用单个节点调用与使用节点列表调用之间的区别?