选择einsum

Python包opt_einsum:优化的einsum可以通过优化表达式的收缩顺序和将许多操作分派到规范的BLAS、cuBLAS或其他专用例程来显著减少类einsum表达式的总体执行时间。优化的einsum对后端是不可知的,它可以处理NumPy、Dask、PyTorch、Tensorflow、CuPy、Sparse、Theano、JAX和Autograd数组,还可以处理任何符合标准API的库。