triton.language.min
- triton.language.min = JITFunction(triton.language.standard:min)
Returns the minimum of all elements in the
input
tensor along the providedaxis
- 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
, asx.min(...)
instead ofmin(x, ...)
.