triton.language.sum¶
- triton.language.sum(input, axis=None, keep_dims=False)¶
Returns the sum of all elements in the
input
tensor along the providedaxis
- 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
This function can also be called as a member function on
tensor
, asx.sum(...)
instead ofsum(x, ...)
.