矢量化加法

Question

假设我有两个 numpy 数组 a[n,3] 和 b[m,3]

如何计算 c[n,m,3] ，而不求助于 for 循环，例如：

c[i,j,:] = a[i,:] + b[j,:]

Answer 1

试试这个：

c = a[:, None, :] + b[None, :, :]