假设我们有两个数组,A和B。现在假设我们想要:
用一种特殊的方法将A和B相乘来创建新的乘积的数组,然后可能
沿特定轴求和这个新数组,和/或
按特定顺序转置数组的轴。...[4, 5, 6, 7],
[8, 9,10,11]])
我们通常如何在NumPy中执行此操作?...一个很好的例子是矩阵乘法,它将行与列相乘,然后对乘积结果求和。对于两个二维数组A和B,矩阵乘法操作可以用np.einsum(‘ij,jk->ik’, A, B)完成。
这个字符串是什么意思?...例如,’ij,jk->ki’为矩阵乘法的转置。
现在,我们已经知道矩阵乘法是如何工作的。...如函数dot和inner经常链接到BLAS例程可以超越einsum在速度方面,tensordot函数也可以与之相比。