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 的维度，以便将维度对齐以进行广播。