< Back

Env

import torch


a = torch.arange(3).reshape((3, 1))
b = torch.arange(2).reshape((1, 2))
a, b 
(tensor([[0],
         [1],
         [2]]),
 tensor([[0, 1]]))

a + b 
tensor([[0, 1],
        [1, 2],
        [2, 3]])

what happened?

a copied its own dim 1, and b copied its own dim 0.

  3D tensor

a = torch.arange(3).reshape((3, 1))
b = torch.arange(2).reshape((1, 2))
a, b 
(tensor([[0],
         [1],
         [2]]),
 tensor([[0, 1]]))

a + b 
tensor([[0, 1],
        [1, 2],
        [2, 3]])

what happened?

a copied its own dim 1, and b copied its own dim 0.