From 35c67f6d33aa82d1ee8bd4fbdc5b925ef560de2a Mon Sep 17 00:00:00 2001
From: Alex Rogozhnikov <iamfullofspam@gmail.com>
Date: Sat, 3 Oct 2020 02:28:51 -0700
Subject: DOC: add references to einops and opt_einsum

Following discussion in mailing list, this PR
adds mentions to third-party libs einops and opt_einsum
to einsum documentation
---
 numpy/core/einsumfunc.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'numpy/core/einsumfunc.py')

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
-- 
cgit v1.2.1