2.1 练习

看下这个答案是怎么计算出来的

# shape不一样且大小不匹配的位置任意有一个张量为1
a = torch.arange(3).reshape((1,3,1))
b = torch.arange(9).reshape((3,1,3))
a, b, a+b

我们一维一维拓展
对a：

• 维度2：
  • a[0][0]为[0]，复制三个之后变成[0,0,0]
  • a[0][1]为[1]，复制三个之后变成[1,1,1]
  • a[0][2]为[2]，复制三个之后变成[2,2,2]
• 维度1：不用复制
• 维度0：
  • a为

  [[[0,0,0],
  [1,1,1],
  [2,2,2]]]
  

  复制三个之后变成

  [
  [[0,0,0],
  [1,1,1],
  [2,2,2]],
  
  [[0,0,0],
  [1,1,1],
  [2,2,2]],
  
  [[0,0,0],
  [1,1,1],
  [2,2,2]]
  ]
  

对b同理操作，最后再相加即可