我想通過簡單地沿一個或多個軸執行相同的加法來添加具有不同維度的兩個數組。非矢量化解決方案:x = np.array([[[1,2],[3,4],[5,6]],[[7,8],[9,0],[1,2]],[[3,4],[5,6],[7,8]],[[9,0],[1,2],[3,4]]]) #shape (4,3,2)y = np.array([[1,2],[3,4],[5,6],[7,8]]) #shape (4,2)ans = np.copy(x)for i in range(x.shape[1]): ans[:,i] += yprint(ans) #shape (4,3,2)我怎樣才能沿著(例如)第二個軸廣播這個?更一般地說,我如何沿任意軸進行此廣播?
添加回答
舉報
0/150
提交
取消
