triton.language.xor_sum

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

返回input张量中所有元素沿指定axis的异或和

Parameters:

• input (Tensor) – 输入值

• axis (int) – 指定进行归约操作的维度。如果为None，则对所有维度进行归约

• keep_dims (bool) - 如果为true，则保留长度为1的缩减维度

此函数也可以作为成员函数在tensor上调用， 形式为x.xor_sum(...)而非 xor_sum(x, ...)。