Check row equality of tensor

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