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 – 是否返回池化索引。默认值:
False