megengine.functional.nn.adaptive_max_pool2d

adaptive_max_pool2d(inp, oshp)

对输入数据进行2D最大池化。

更多信息参见 MaxPool2d。

参数:

• inp (Tensor) – 输入张量。

• oshp (Union[Tuple[int, int], int, Tensor]) – (OH, OW) 输出形状的大小。

返回类型:

Tensor

返回:

输出张量。