现在假设我们想要:
用一种特殊的方法将A和B相乘来创建新的乘积的数组,然后可能
沿特定轴求和这个新数组,和/或
按特定顺序转置数组的轴。...[4, 5, 6, 7],
[8, 9,10,11]])
我们通常如何在NumPy中执行此操作?...对于两个二维数组A和B,矩阵乘法操作可以用np.einsum(‘ij,jk->ik’, A, B)完成。
这个字符串是什么意思?想象’ij,jk->ik’在箭头->处分成两部分。...左侧部分标记输入数组的轴:’ij’标记A和’jk’标记B。字符串的右侧部分用字母“ik”标记单个输出数组的轴。也就是说,我们正在传入两个二维数组,获取一个新的二维数组。...如函数dot和inner经常链接到BLAS例程可以超越einsum在速度方面,tensordot函数也可以与之相比。