3-9广播机制

import torch

a = torch.rand(2, 2)
b = torch.rand(1, 2)
# a, 2*1
# b, 1*2
# c, 2*2
# 2*4*2*3
c = a + b
print(a)
print(b)
print(c)
print(c.shape)

tensor([[0.3246, 0.0065],
        [0.5726, 0.3850]])
tensor([[0.3750, 0.1614]])
tensor([[0.6996, 0.1680],
        [0.9476, 0.5464]])
torch.Size([2, 2])