triton.language.min

triton.language.min(input, axis=None, return_indices=False, return_indices_tie_break_left=True, keep_dims=False)

Returns the minimum of all elements in the input tensor along the provided axis

Parameters:
  • input (Tensor) – the input values

  • axis (int) – the dimension along which the reduction should be done. If None, reduce all dimensions

  • keep_dims (bool) – if true, keep the reduced dimensions with length 1

  • return_indices (bool) – if true, return index corresponding to the minimum value

  • return_indices_tie_break_left (bool) – if true, in case of a tie (i.e., multiple elements have the same minimum value), return the left-most index for values that aren’t NaN

This function can also be called as a member function on tensor, as x.min(...) instead of min(x, ...).