mindspore.mint.nn.AvgPool2d

class mindspore.mint.nn.AvgPool2d(kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True, divisor_override=None)

Apply a 2-D average pooling over an input tensor which can be regarded as a composition of 2-D input planes.

For details, please refer to mindspore.mint.nn.functional.avg_pool2d().

Supported Platforms:

Ascend

Examples

>>> import mindspore
>>> input = mindspore.tensor(mindspore.mint.arange(1 * 3 * 3 * 4).reshape(1, 3, 3, 4), mindspore.float32)
>>> net = mindspore.mint.nn.AvgPool2d(kernel_size=2, stride=1)
>>> output = net(input)
>>> print(output.shape)
(1, 3, 2, 3)