[4, 5, 6, 7],
[8, 9,10,11]])
我们通常如何在NumPy中执行此操作?...要了解输出数组的计算方法,请记住以下三个规则:
在输入数组中重复的字母意味着值沿这些轴相乘。乘积结果为输出数组的值。
在本例中,我们使用字母j两次:A和B各一次。这意味着我们将A每一行与B每列相乘。...这只在标记为j的轴在两个数组中的长度相同(或者任一数组长度为1)时才有效。
输出中省略的字母意味着沿该轴的值将相加。
在这里,j不包含在输出数组的标签中。...注意,由于np.einsum(‘ij,jk->ik’, A, B)函数不构造3维数组然后求和,它只是将总和累加到2维数组中。
一些简单的操作
这就是我们开始使用einsum时需要知道的全部内容。...现在,我们A和B是与之兼容形状的两个二维数组:
?
当处理大量维度时,别忘了einsum允许使用省略号语法’…’。