triton.language.argmin

triton.language.argmin(input, axis, tie_break_left=True, keep_dims=False)

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

Parameters:
  • input – the input values

  • axis – the dimension along which the reduction should be done

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

  • tie_break_left – if true, return the left-most indices in case of ties for values that aren’t NaN

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