PGL
PGL copied to clipboard
关于文档中recv的建议
文档中recv要求arg的msg: A LodTensor or a dictionary of LodTensor whose batch_size is equals to the number of unique dst nodes. 但是send返回的msg里面的东西的shape都是msg [num_edges, dims ]。这里好像没统一。
另外貌似文档中对send的描述也不对,实际使用时只能返回dict,不能返回tensor