opt_einsum::high_voltage:通过收缩顺序优化优化 NumPy、Tensorflow、Dask 等中的 ei...

优化的 Einsum 优化的 Einsum：张量收缩顺序优化器 优化的 einsum 可以通过优化表达式的收缩顺序并将许多操作分派到规范 BLAS 来显着减少类 einsum 表达式（例如， 、 、 、 ）的整体执行时间， cuBLAS 或其他专门的例程。 优化的 einsum 与后端无关，可以处理 NumPy、Dask、PyTorch、Tensorflow、CuPy、Sparse、Theano、JAX 和 Autograd 数组以及可能符合标准 API 的任何库。 有关更多信息，请参阅。 示例用法 函数通常可以作为einsum函数的替代品，无需进一步更改代码，同时提供卓越的性能。 这里，张量收缩是在优化和不优化的情况下进行的： import numpy as np from opt_einsum import contract N = 10 C = np . random