diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2020-10-04 12:21:51 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-10-04 12:21:51 -0600 |
| commit | 2dbfb458fa3e96203f93118a9d142a6bedca9271 (patch) | |
| tree | 27ce8054f265f4d5a0d243f3cadcb12b0d802818 | |
| parent | 81829432b3a00ae8ebe46e53d181810b462f0024 (diff) | |
| parent | 35c67f6d33aa82d1ee8bd4fbdc5b925ef560de2a (diff) | |
| download | numpy-2dbfb458fa3e96203f93118a9d142a6bedca9271.tar.gz | |
Merge pull request #17432 from arogozhnikov/add-einops-opt_einsum-reference
DOC: add references to einops and opt_einsum
| -rw-r--r-- | numpy/core/einsumfunc.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/numpy/core/einsumfunc.py b/numpy/core/einsumfunc.py index f65f4015c..e0942beca 100644 --- a/numpy/core/einsumfunc.py +++ b/numpy/core/einsumfunc.py @@ -1062,6 +1062,17 @@ def einsum(*operands, out=None, optimize=False, **kwargs): -------- einsum_path, dot, inner, outer, tensordot, linalg.multi_dot + einops: + similar verbose interface is provided by + `einops <https://github.com/arogozhnikov/einops>`_ package to cover + additional operations: transpose, reshape/flatten, repeat/tile, + squeeze/unsqueeze and reductions. + + opt_einsum: + `opt_einsum <https://optimized-einsum.readthedocs.io/en/stable/>`_ + optimizes contraction order for einsum-like expressions + in backend-agnostic manner. + Notes ----- .. versionadded:: 1.6.0 |