Given tensor A with shape(5x1) and tensor B with shape(1x5)
In pytorch,
import torch
torch.eq(A, B) will produce C with shape(5x1), but this is not my expectation.
Is there an equivalent of “tensorflow.equal(A, B)” in pytorch that produce a tensor with shape of (5x5) ?
thanks,
metas