CMD + K

opt_einsum

Community

Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.

Installation

To install this package, run one of the following:

Conda
$conda install rocketce::opt_einsum

Usage Tracking

3.3.0
1 / 8 versions selected
Downloads (Last 6 months): 0

Description

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

About

Summary

Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.

Last Updated

Jan 11, 2023 at 04:35

License

MIT

Total Downloads

1.5K

Supported Platforms

noarch