为了账号安全,请及时绑定邮箱和手机立即绑定

沿任意轴的 numpy 广播加法

沿任意轴的 numpy 广播加法

繁花不似锦 2021-08-11 21:36:56
我想通过简单地沿一个或多个轴执行相同的加法来添加具有不同维度的两个数组。非矢量化解决方案: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)我怎样才能沿着(例如)第二个轴广播这个?更一般地说,我如何沿任意轴进行此广播?
查看完整描述

1 回答

  • 1 回答
  • 0 关注
  • 190 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信