torch_geometric 笔记:global_mean_pool
生活随笔
收集整理的這篇文章主要介紹了
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的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: NTU 课程笔记: MAS714 141
- 下一篇: pytorch 笔记: 复现论文 Sto