PyTorch Geometric(PyG) Convolutional Layers简介

class MessagePassing(aggr='add', flow='source_to_target', node_dim=0)

用于生成message passing层的基类,
在这里插入图片描述
□表示可微的排列不变性函数,例如 sum,mean或max,以及γΘ 和 ϕΘ表示可区分的功能,例如MLP。
参数:
aggr (string, optional) – 所用的聚合方法 (“add”, “mean”, “max” or None). (default: “add”)
flow (string, optional) – 消息传递的流向 (“source_to_target” or “target_to_source”). (default: “source_to_target”)
node_dim (int, optional) – 传播所沿的轴. (default: 0)

原创文章 32 获赞 4 访问量 7630

猜你喜欢

转载自blog.csdn.net/yrwang_xd/article/details/106100882