triton.experimental.gluon.language.reduce_or
- triton.experimental.gluon.language.reduce_or(input, axis, keep_dims=False)
Returns the reduce_or of all elements in the
inputtensor along the providedaxisThe reduction operation should be associative and commutative.
- 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
This function can also be called as a member function on
tensor, asx.reduce_or(...)instead ofreduce_or(x, ...).