通过矩阵广播张量除法

Question

我试图计算沿第三轴的600x400x100张量A除以600x400矩阵B的值。您可以想象一个100帧的视频剪辑，每个600x400帧正在按元素划分。我的代码就像：

A/B

错误消息显示：
ValueError：操作数不能与形状（600,400,129）（600,400）一起广播
我的代码有什么问题？

Answer 1

只需向B添加一个空轴，使其为(600, 400, 1)。 Broadcasting将接管其余部分。

A = np.random.rand(600, 400, 100)
B = np.random.rand(600, 400)

A / B[..., None]