torch_geometric 笔记:global_mean_pool

对全图的点嵌入(node embedding)进行池化操作,返回一个图嵌入(graph embedding)

global_mean_pool(
    x, 
    batch, 
    size=None)[
x (torch.tensor) 输入的点嵌入
batch (torch.tensor 指示点属于哪个图
size (optional 图的数量,可以自动推断得到

你可能感兴趣的:(pytorch学习,深度学习)