< Back

Env

import torch

  ==

X = torch.arange(12, dtype=torch.float32).reshape((3,4))
Y = torch.tensor([[2.0, 1, 4, 3], [1, 2, 3, 4], [4, 3, 2, 1]])
Z = X == Y
X, Y 
(tensor([[ 0.,  1.,  2.,  3.],
         [ 4.,  5.,  6.,  7.],
         [ 8.,  9., 10., 11.]]),
 tensor([[2., 1., 4., 3.],
         [1., 2., 3., 4.],
         [4., 3., 2., 1.]]))

Z 
tensor([[False,  True, False,  True],
        [False, False, False, False],
        [False, False, False, False]])

  >

Z = X > Y
Z 
tensor([[False, False, False, False],
        [ True,  True,  True,  True],
        [ True,  True,  True,  True]])

  <

Z = X < Y
Z 
tensor([[ True, False,  True, False],
        [False, False, False, False],
        [False, False, False, False]])