nvidia.dali.experimental.dynamic.reductions.max#

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

Gets maximal input element 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.

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

On this page