nvidia.dali.experimental.dynamic.reductions.mean#

nvidia.dali.experimental.dynamic.reductions.mean(
input,
/,
*,
batch_size=None,
device=None,
axes=None,
axis_names=None,
dtype=None,
keep_dims=None,
)#

Gets mean of elements along provided axes.

Supported backends

  • ‘cpu’

  • ‘gpu’

Parameters:

input (Tensor/Batch) – Input to the operator.

Keyword Arguments:

  • axes (int or list of int, optional) –

    Axis or axes along which reduction is performed.

    Accepted range is [-ndim, ndim-1]. Negative indices are counted from the back.

    Not providing any axis results in reduction of all elements.

  • axis_names (layout str, optional) –

    Name(s) of the axis or axes along which the reduction is performed.

    The input layout is used to translate the axis names to axis indices, for example axis_names="HW" with input layout “FHWC” is equivalent to specifying axes=[1,2]. This argument cannot be used together with axes.

  • dtype (nvidia.dali.types.DALIDataType, optional) – Output data type. This type is used to accumulate the result.

  • keep_dims (bool, optional, default = False) – If True, maintains original input dimensions.

On this page