triton.language.xor_sum

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

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

参数:

• input (张量) – 输入值

• axis (int) – 应执行归约的维度。如果为 None，则归约所有维度

• keep_dims (bool) – 如果为 true，则保留长度为 1 的归约维度

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