Numpy Broadcasting Addition

时间:2018-04-01 02:27:52

标签: python numpy numpy-broadcasting

如果我有一个y:64 x 400 x 20的数组,另一个数组x:64 x 400,那么做x + y的最有效方法是什么,我在其中广播每个元素的添加x进入y中的每个64x400口袋(所以将每个x [i] [j]添加到y [i] [j] [:]),结果也是64x400x20?

1 个答案:

答案 0 :(得分:1)

x[..., None] + y

x的末尾添加一个额外的长度为1的维度,以便为广播排列尺寸。