1 max_pool?

1.1 函數介紹

torch_geometric.nn.max_pool(cluster,?data,transform=None)

????????對由torch_geometricy .data給出的圖形進行池化和粗化。

????????數據對象根據集群cluster中定義的集群。同一集群中的所有節點將表示為一個節點。最終節點特征由同一簇內所有節點的特征最大值定義，節點位置平均，邊的index定義為同一簇內所有節點的邊index的并集。

1.2 參數說明

cluster?(LongTensor)?	簇向量，每一個維度表示了一個點屬于哪個簇
data?(Data)?	torch_geometric的data 對象
transform?(callable,?optional)	一個函數/轉換，接受粗化和池化的torch_geometry .data。數據對象，并返回轉換后的版本。

?返回torch_geometric的data 對象

1.3 舉例說明

假如我們一開始的data 為：

Batch(x=[9893, 1], edge_index=[2, 34637], y=[9893, 1], batch=[9893], ptr=[2]) from torch_geometric.nn import max_poolcluster = graclus(data.edge_index, num_nodes=x.shape[0]) cluster #tensor([ 0, 1, 1, ..., 9890, 9891, 9892]) #第i維表示第i個點在以第幾個點為中心點的簇中data_c = max_pool(cluster, data) data_c #Batch(x=[5863, 1], edge_index=[2, 21983], batch=[5863]) #分成了5863 個cluster

1.3.1 mini-batch 的max_pool

在mini_batch的話，需要這樣寫：

data_c = max_pool(cluster, Data(x=data.x, batch=data.batch, edge_index=data.edge_index)) data

2 max_pool_x

對一個cluster中中的x的特征進行最大池化操作

max_pool_x(cluster, x, batch, size: Optional[int] = None)

注意和max_pool的區別

max_pool 返回的是data，max_pool_x返回的是Tensor

max_pool 相當于max_pool_x的基礎上，再對圖的邊進行了修改合并操作

《新程序員》：云原生和全面數字化實踐50位技術專家共同創作，文字、視頻、音頻交互閱讀

總結

以上是生活随笔為你收集整理的torch_geometric笔记：max_pool 与max_pool_x的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。