Table Of Contents
Table Of Contents

mxnet.ndarray.contrib.AdaptiveAvgPooling2D

mxnet.ndarray.contrib.AdaptiveAvgPooling2D(data=None, output_size=_Null, out=None, name=None, **kwargs)

Applies a 2D adaptive average pooling over a 4D input with the shape of (NCHW). The pooling kernel and stride sizes are automatically chosen for desired output sizes.

  • If a single integer is provided for output_size, the output size is (N x C x output_size x output_size) for any input (NCHW).

  • If a tuple of integers (height, width) are provided for output_size, the output size is (N x C x height x width) for any input (NCHW).

Defined in src/operator/contrib/adaptive_avg_pooling.cc:L214

Parameters

  • data (NDArray) – Input data

  • output_size (Shape(tuple), optional, default=[]) – int (output size) or a tuple of int for output (height, width).

  • out (NDArray, optional) – The output NDArray to hold the result.

Returns

out – The output of this function.

Return type

NDArray or list of NDArrays

Previous
mxnet.ndarray.contrib.quantize
Next
mxnet.ndarray.contrib.BilinearResize2D