triton.language.xor_sum

triton.language.xor_sum(input, axis=None, keep_dims=False)

Returns the xor sum of all elements in the input tensor along the provided axis

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, as x.xor_sum(...) instead of xor_sum(x, ...).