[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时需要知道的全部内容。