PyTorch: compare three tensors?
AFAIK, the tensor is basically a NumPy array bound to the device. If not too expensive for your application and you can afford to do it on CPU, you can simply convert it to NumPy and do what you need with the comparison.
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.