triton.language.max

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

Returns the maximum 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

• return_indices – if true, return index corresponding to the maximum value

• return_indices_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.max(...) instead of max(x, ...).