Function min

python/tvm/topi/reduction.py:145–167

Minimum of array elements over a given axis or a list of axes Parameters ---------- data : tvm.te.Tensor The input tvm tensor axis : None or int or tuple of int Axis or axes along which a minimum operation is performed. The default, axis=None, will find the

(data, axis=None, keepdims=False)

def min(data, axis=None, keepdims=False):
    """Minimum of array elements over a given axis or a list of axes

    Parameters
    ----------
    data : tvm.te.Tensor
        The input tvm tensor

    axis : None or int or tuple of int
        Axis or axes along which a minimum operation is performed.
        The default, axis=None, will find the minimum element from all of the elements of the
        input array. If axis is negative it counts from the last to the first axis.

    keepdims : bool
        If this is set to True, the axes which are reduced are left in the result as dimensions
        with size one.
        With this option, the result will broadcast correctly against the input array.

    Returns
    -------
    ret : tvm.te.Tensor
    """
    return cpp.min(data, axis, keepdims)


def argmax(data, axis=None, keepdims=False, select_last_index=False):