def adaptive_avg_pool2d(self, output_size):
r"""
Applies a 2D adaptive average pooling over an input signal composed of
several input planes.
See :class:`~torch.nn.AdaptiveAvgPool2d` for details and output shape.
Args:
output_size: the target output size (single integer or
double-integer tuple)
"""
resized_input, args, kwargs = adaptive_pool2d_helper(self,
output_size,
reduction="mean")
return resized_input.avg_pool2d(*args, **kwargs)
def adaptive_max_pool2d(self, output_size, return_indices=False):
r"""Applies a 2D adaptive max pooling over an input signal composed of
several input planes.
See :class:`~torch.nn.AdaptiveMaxPool2d` for details and output shape.
Args:
output_size: the target output size (single integer or
double-integer tuple)
return_indices: whether to return pooling indices. Default: ``False``
"""
resized_input, args, kwargs = adaptive_pool2d_helper(
self, output_size, reduction="max"
)
return resized_input.max_pool2d(*args, **kwargs, return_indices=return_indices)
