torch.equal
torch.equal(input, other) -> bool
"True" if two tensors have the same size and elements, "False"
   otherwise.
Example:
  >>> torch.equal(torch.tensor([1, 2]), torch.tensor([1, 2]))
  True
