对全图的点嵌入(node embedding)进行池化操作,返回一个图嵌入(graph embedding)
global_mean_pool(
x,
batch,
size=None)[
x (torch.tensor) | 输入的点嵌入 |
batch (torch.tensor) | 指示点属于哪个图 |
size (optional) | 图的数量,可以自动推断得到 |
对全图的点嵌入(node embedding)进行池化操作,返回一个图嵌入(graph embedding)
global_mean_pool(
x,
batch,
size=None)[
x (torch.tensor) | 输入的点嵌入 |
batch (torch.tensor) | 指示点属于哪个图 |
size (optional) | 图的数量,可以自动推断得到 |