If I have a N, nB dimensional tensor, how do I check that each 1 of N ‘rows’ of the tensor are equal?
If you would like to check if all rows are equal, you could check the first row against all others:
Alternatively, you could call
torch.unique on your tensor with
dim=0 and check dim0 for the shape:
torch.unique(x, dim=0).size(0) == 1