直接看代码:
>>> a=torch.randn(1,2,3) #这个tensor的size为(1,2,3)
>>> a
tensor([[[ 0.6591, -0.2453, 1.2475],
[ 0.6525, -1.8686, 0.0861]]])
>>> b=torch.randn(3,1,1) #这个tensor的size为(3,1,1)
>>>> b
tensor([[[ 0.8488]],
[[-1.6082]],
[[ 0.1244]]])
>>> a+b #相加后的size为(3,2,3) 注意相加的顺序!
tensor([[[ 1.5079, 0.6035, 2.0962],
[ 1.5013, -1.0198, 0.9349]],
[[-0.9491, -1.8535, -0.3607],
[-0.9556, -3.4767, -1.5221]],
[[ 0.7835, -0.1209, 1.3719],
[ 0.7770, -1.7441, 0.2105]]])