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