Tensor中的加法

輸入如下代碼：

import torch
a=torch.zeros(2,1)
b=torch.ones(1,3)
c=a+b
print("a=",a)
print("b=",b)
print("c=",c)

輸出爲：

a= tensor([[0.],
        [0.]])
b= tensor([[1., 1., 1.]])
c= tensor([[1., 1., 1.],
        [1., 1., 1.]])

加法的具體計算過程如下：

1、將a的形狀擴展爲（2，3）

a= tensor([[0., 0., 0.],
        [0., 0., 0.]])

2、將b的形狀擴展爲（2，3）

b= tensor([[1., 1., 1.],
        [1., 1., 1.]])

3、然後兩個tensor對應位置的值相加得到最終結果

c= tensor([[1., 1., 1.],
        [1., 1., 1.]])

注意事項

如下代碼運行會出錯，因爲兩個tensor的形狀不匹配

import torch
a=torch.zeros(2,1)
b=torch.ones(3,1)
c=a+b

參考文章：https://blog.csdn.net/weixin_40952784/article/details/92575851