具有多个索引的矩阵的 np.einsum 乘法？答案

【问题标题】：np.einsum multiplication of matrices with several indices?具有多个索引的矩阵的 np.einsum 乘法？
【发布时间】：2021-08-19 00:31:47
【问题描述】：

给定 m x n 矩阵 A 和 n x r 矩阵 B 如何用np.einsum 符号写出下面的公式？

f(i) = \sum_{j,k} a_ij * b_jk

如果添加 r x p 矩阵 C，np.einsum 会发生什么变化？

f(i) = \sum_{j,k,l} a_ij * b_jk * c_kl

【问题讨论】：

【解决方案1】：

@Sengiley 尽管您已经回答了自己的问题。允许广播的更通用版本是：

np.einsum('...ij,jk,kl->...i', a, b, c)

【讨论】：