An extension of np.einsum where capital letters serve as multiple ellipses.
Project description
keinsum
An extension of np.einsum where capital letters serve as multiple ellipses, see Einsum Visualized for details.
Installation:
pip install keinsum
Usage
from keinsum import keinsum
keinsum('Ik,kJ', a, b)`
is the same as np.einsum('ijk,klm', a, b)
where a
and b
are 3D arrays.
Testing
Run pytest
in the project root.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
keinsum-0.4.tar.gz
(2.6 kB
view hashes)
Built Distribution
keinsum-0.4-py3-none-any.whl
(2.5 kB
view hashes)