假设我们有两个数组,A和B。现在假设我们想要:
用一种特殊的方法将A和B相乘来创建新的乘积的数组,然后可能
沿特定轴求和这个新数组,和/或
按特定顺序转置数组的轴。...[4, 5, 6, 7],
[8, 9,10,11]])
我们通常如何在NumPy中执行此操作?...这提供了一种变量的方式标记我们不大感兴趣的轴,例如np.einsum(‘…ij,ji->…’, a, b),仅将a的最后两个轴与2维数组b相乘。
注意事项
本节说一些使用该函数时要注意的东西。...如果你使用的是更有限的数据类型,则可能会出现意外结果:
>>> a= np.ones(300, dtype=np.int8)
>>> np.sum(a)# correct result
300
>>>...如函数dot和inner经常链接到BLAS例程可以超越einsum在速度方面,tensordot函数也可以与之相比。