python - 如何在 numpy 中进行类似二元运算的操作

Question

我有两个二维数组 A 和 B。我想得到一个 3-D 数组 C，它与 A 和 B 的关系是:

C_mnl=A_mn*B_ml

我怎样才能在 numpy 中优雅地做到这一点？

Answer 1

numpy.einsum可以这样做:

a = np.arange(6).reshape(3,2)  # a.shape = (3, 2)
b = np.arange(12).reshape(3,4)  # b.shape = (3, 4)

c = np.einsum('mn,ml->mnl', a, b)  # c.shape = (3, 2, 4)