numpy中的快速外张量积

Question

我有两个numpy数组:

x of shape ((d1,...,d_m)) 
y of shape ((e_1,...e_n)) 

我想形成外张量产品,即numpy数组

z of shape ((d1,...,d_m,e_1,...,e_n))

这样的

z[i_1,...,i_n,i_{n+1}...,i_{m+n}] == x[i_1,...i_m]*y[i_{m+1},...,i_{m+n}]

我必须多次执行上面的外部乘法,所以我想尽可能加快速度.

Answer 1

你想要np.multiply.outer:

z = np.multiply.outer(x, y)