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)