如何在Python / numpy中执行张量生成？

Question

我有两个numpy 2D数组，我想执行此操作：

a_ij * b_ik = c_ijk

如何使用numpy制作它？

Answer 1

einsum为此任务量身定制

a_ij * b_ik = c_ijk

c = np.einsum('ij,ik->ijk', a, b)

===================

但正如Divakar所示，没有暗示总和，所以普通乘法也是如此，a[...,None]*b[:,None,:]。