chainerx.average_pool

chainerx.average_pool(x, ksize, stride=None, pad=0, pad_mode='ignore')

Spatial average pooling function.

This acts similarly to conv(), but it computes the average of input spatial patch for each channel without any parameter instead of computing the inner products.

Parameters

  • x (ndarray) – Input array.

  • ksize (int or tuple of ints) – Size of pooling window. ksize=k and ksize=(k, k, ..., k) are equivalent.

  • stride (int or tuple of ints or None) – Stride of pooling applications. stride=s and stride=(s, s, ..., s) are equivalent. If None is specified, then it uses same stride as the pooling window size.

  • pad (int or tuple of ints) – Spatial padding width for the input array. pad=p and pad=(p, p, ..., p) are equivalent.

  • pad_mode ({'zero', 'ignore'}) –

    Specifies how padded region is treated.

    • ’zero’ – the values in the padded region are treated as 0

    • ’ignore’ – padded region is ignored (default)

Returns

Output array.

Return type

ndarray

Note

During backpropagation, this function propagates the gradient of the output array to the input array x.

Note

In cuda backend, only 2 and 3 dim arrays are supported as x because cuDNN pooling supports 2 and 3 spatial dimensions.