class TopKPooling(in_channels, ratio=0.5, min_score=None, multiplier=1, nonlinearity=<built-in method tanh of type object>)
参数:
in_channels (int) – 输入通道数，也就是x的shape[1].
ratio (float) – 池化比率，用来计算 k=⌈ratio⋅N⌉. 如果 min_score 不是 None那么这个值将被忽略. (比率默认: 0.5)
本文作者注：这个比率是指图经过池化后剩余的节点数占池化之前的节点数的比例
min_score (float, optional) – 最小节点得分α~ 用于计算池化的节点的索引$\mathbf{i} = \mathbf{y}_i > \tilde{\alpha} $(default: None)
multiplier (float, optional) – 合并后要素乘以的系数。这对于大型图形以及使用min_score时可能很有用 (default: 1)
nonlinearity (torch.nn.functional, optional) – 使用的非线性函数. (default: torch.tanh)
