python - 在这种情况下，numpy 广播如何工作？

Question

有人可以逐步解释在这种情况下 numpy 广播是如何工作的吗？

a = np.ones((2,3))
b = np.ones((2,1,3))
c = a-b

a.shape
(2, 3)
b.shape
(2, 1, 3)
c.shape
(2, 2, 3)

标签： pythonnumpy

Answer 1

参考这个页面，它说 numpy 在具有较低等级的张量前面加上 1，所以在这种情况下，我们有 a.shape = [1,2,3]

沿轴 1平铺a以获得a.shape=[2,2,3]

沿轴 2平铺b以获得b.shape=[2,2,3]

当尺寸相同时，减去