torch.eq# torch.eq(input, other, *, out=None) → Tensor# Computes element-wise equality The second argument can be a number or a tensor whose shape is broadcastable with the first argument. Parameters input (Tensor) – the tensor to compare other (Tensor or float) – the tensor or value to compare Keyword Arguments out (Tensor, optional) – the output tensor. Returns A boolean tensor that is True where input is equal to other and False elsewhere Example: >>> torch.eq(torch.tensor([[1, 2], [3, 4]]), torch.tensor([[1, 1], [4, 4]])) tensor([[ True, False], [False, True]])