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
