torch_geometric 笔记：global_mean_pool
Posted UQI-LIUWJ
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了torch_geometric 笔记：global_mean_pool相关的知识，希望对你有一定的参考价值。
对全图的点嵌入(node embedding)进行池化操作，返回一个图嵌入（graph embedding)
global_mean_pool(
x,
batch,
size=None)[| x (torch.tensor) | 输入的点嵌入 |
| batch (torch.tensor) | 指示点属于哪个图 |
| size (optional) | 图的数量，可以自动推断得到 |
以上是关于torch_geometric 笔记：global_mean_pool的主要内容，如果未能解决你的问题，请参考以下文章
torch_geometric笔记：nn. graclus （图点分类）
pytorch 笔记：torch_geometric 创建一张图
torch_geometric 笔记： 数据集Cora &简易 GNN