python - numexpr を使用したアインシュタイン総和の評価

Question

私は現在、 を使用して大規模な numpy 配列の乗算を行っており、問題numpy.einsumに直面していMemoryErrorます。そのため、可能な限り式を評価しようとしていnumexprます。私が理解する限りでは：

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

技術的には

numexpr.evaluate("a*b")

ただし、他の場合はそれほど単純ではないようです。次の 3 つの式があるとします。

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

を使用してこれら 3 つを達成する同等の方法は何でしょうnumexprか?