python - Numpy 广播添加

Question

如果我有一个 y: 64 x 400 x 20 数组和另一个数组 x: 64 x 400，执行 x + y 的最有效方法是什么，我将 x 中每个元素的添加广播到每个元素中y 中的 64x400 个口袋(因此将每个 x[i][j] 添加到 y[i][j][:] 中)，这样结果也是 64x400x20？

Answer 1

x[..., None] + y

在 x 末尾添加额外的长度为 1 的尺寸，以使尺寸对齐以进行广播。