Pytorch中的广播机制

广播机制：张量参数可以自动扩展为相同大小

广播机制需要满足两个条件：

        ① 每个张量至少有一个维度

        ② 满足右对齐

案例：

a = torch.rand(2, 1)
b = torch.rand(3)
c = torch.add(a, b)
print(a)
print(b)
print(c)

"""
tensor([[0.0121],
        [0.9789]])

tensor([0.1547, 0.4282, 0.2488])

tensor([[0.1668, 0.4403, 0.2609],
        [1.1336, 1.4071, 1.2277]])

"""