评价此页

torch.nn.functional.adaptive_max_pool2d#

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

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

请参阅 AdaptiveMaxPool2d 以获取详细信息和输出形状。

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

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