numpy - 将 sum(...) 转换为 numpy

Question

我想sum(ai*np.exp(bi*r2) for ai,bi in zip(a,b))完全在 numpy 中完成，其中a和b是一维数组，并且r2可以是任何形状，结果的形状为r2. 我的广播福太弱了！

标签： numpy

Answer 1

尝试

  np.sum(a*np.exp(b*r2[...,None], axis=-1)

这个想法是在最后以 a,b 维度进行广播。