torch.nn.functional.adaptive_max_pool2d

torch.nn.functional.adaptive_max_pool2d(input, output_size, return_indices=False)

在由多个输入平面组成的输入信号上应用二维自适应最大池化。

有关详细信息和输出形状，请参见AdaptiveMaxPool2d。

参数

• output_size – 目标输出大小（单个整数或双整数元组）

• return_indices – 是否返回池化索引。默认值：False