Numpy 矩阵张量除的广播

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

A/B 错误信息显示： ValueError:操作数无法与形状600400129 600400一起广播 我的代码怎么了？

只需在B上添加一个空轴，使其为600、400、1。我会接管剩下的

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

A / B[..., None]

---

请使用降价来修复您的标记。谢谢！我通过将B重塑为600x400x1解决了这个问题。A/B[…，无]也有效@查尔斯：如果这个答案帮助你解决问题，请考虑一下。