python下:列表!=数组!=矩阵
# python列表
[1,2]+[3,4]=[1,2,3,4]
[1,2] * 2 = [1,2,1,2]#不是[2,4]
# np数组
np.array([1,2])+np.array([1,2])=array([2, 4])
# tensor矩阵
torch.tensor([1,2])+torch.tensor([1,2])=tensor([2, 4])
# broadcast
a = torch.arange(12).reshape(2,2,3)
b = torch.arange(6).reshape(2,3)
a
b
a+b
broadcast结果:
tensor([[[ 0, 1, 2],
[ 3, 4, 5]],
[[ 6, 7, 8],
[ 9, 10, 11]]])
tensor([[0, 1, 2],
[3, 4, 5]])
tensor([[[ 0, 2, 4],
[ 6, 8, 10]],
[[ 6, 8, 10],
[12, 14, 16]]])