是什么einsum呢
使用einsum函数,我们可以使用爱因斯坦求和约定(Einstein summation convention)在NumPy数组上指定操作。
假设我们有两个数组,A和B。...[4, 5, 6, 7],
[8, 9,10,11]])
我们通常如何在NumPy中执行此操作?...如何使用einsum
关键是为输入数组的轴和我们想要输出的数组选择正确的标签。
函数使我们可以选择两种方式之一执行此操作:使用字符串或使用整数列表。...对于两个二维数组A和B,矩阵乘法操作可以用np.einsum(‘ij,jk->ik’, A, B)完成。
这个字符串是什么意思?想象’ij,jk->ik’在箭头->处分成两部分。...知道如何将不同的轴相乘,然后如何对乘积求和,我们可以迅速而简单地表达许多不同的操作。这使我们可以相对容易地将问题推广到更高维度。例如,我们不必插入新的轴或转置数组以使它们的轴正确对齐。