opt_einsum
Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
To install this package, run one of the following:
Einsum is a very powerful function for contracting tensors of arbitrary dimension and index. However, it is typically only optimized to contract two terms at a time resulting in non-optimal scaling. This package optimizes the contraction order for arbitrarily large speedups. See the docs for more information: http://optimized-einsum.readthedocs.io
Summary
Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
Last Updated
Jun 8, 2021 at 16:31
License
MIT
Total Downloads
368.2K
Supported Platforms
GitHub Repository
https://github.com/dgasmith/opt_einsum