numpy - 用矩阵广播张量除法

Question

我试图计算 600x400x100 张量 A 除以第 3 轴的 600x400 矩阵 B 的值。你可以想象一个 100 帧的视频剪辑，每个 600x400 帧都在进行元素划分。我的代码是这样的：

A/B

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

标签： numpy

Answer 1

只需在 中添加一个空轴B，即可(600, 400, 1)。广播将接管其余部分。

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

A / B[..., None]