triton.language.sum

triton.language.sum(input, axis=None)

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