当输入是 DGL 中的列表时,DGLGraph.pull 会出错

问题描述

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,dtype=torch.float32) 的列。

是否有任何文档可以理解使用单个节点调用与使用节点列表调用间的区别?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)