使用numexpr评估爱因斯坦求和

时间:2018-05-11 21:01:57

标签: python arrays numpy numpy-einsum numexpr

我目前正在使用numpy.einsum处理大型numpy数组乘法,并且一直面临MemoryError问题。这就是为什么我试图用numexpr来尽可能地评估表达式。据我了解:

np.einsum('ij,j -> ij', a, b)

在技术上与

相同
numexpr.evaluate("a*b")

然而,在其他情况下似乎并不那么简单。假设我有三个表达式:

np.einsum('i, j -> ij', a, b)
np.einsum('ij, i -> ij', a, b)
np.einsum(ijk, kl -> ijkl', a, b)

使用numexpr实现这三者的等效方法是什么?

0 个答案:

没有答案