数据操作-广播机制

在之前运算中,都是在相同形状的两个张量上执行按元素操作。在某些情况下,我们仍然可以通过调用 广播机制 (broadcasting mechanism) 来执行按元素操作

1、通过适当复制元素来扩展一个或两个数组,以便在转换之后,两个张量具有相同的形状

2、对生成的数组执行按元素操作。

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 分别是 3×1和 1×2 矩阵,如果直接相加,它们的形状不匹配。我们将两个矩阵广播为一个更大的 3×2矩阵,如下所示:矩阵 a将复制列,矩阵 b将复制行,然后再按元素相加。

c=a + b
print(a)
print(b)
print(c)

#输出结果

tensor([[0],
        [1],
        [2]])
tensor([[0, 1]])
tensor([[0, 1],
        [1, 2],
        [2, 3]])

  

上一篇:shape与reshape


下一篇:4:张量操作