PyTorch: compare three tensors?

I have three boolean mask tensors that I want to create a boolean mask that if the value matches in three tensors then it is 1, else 0.

I tried torch.where(A == B == C, 1, 0), but it doesn't seem to support such.

https://stackoverflow.com/questions/67444564/pytorch-compare-three-tensors May 08, 2021 at 02:04PM