评价此页

torch.nn.functional.adaptive_max_pool2d#

torch.nn.functional.adaptive_max_pool2d(input, output_size, return_indices=False)[源代码]#

对由多个输入平面组成的输入信号应用 2D 自适应最大池化。

详情和输出形状请参阅 AdaptiveMaxPool2d

参数
  • output_size – 目标输出大小 (单个整数或双整数元组)

  • return_indices – whether to return pooling indices. Default: False