Fix tensordot broadcasting rules (#419)

asmeurer · web-flow · commit c5808f2b173e · 2022-04-21T03:02:48.000-07:00
* Fix tensordot broadcasting rules

The noncontracted axes do not broadcast together and do not need to be
broadcast compatible.

* Replace a "should" with "must"
diff --git a/spec/API_specification/signatures/linear_algebra_functions.py b/spec/API_specification/signatures/linear_algebra_functions.py
@@ -61,7 +61,7 @@ def tensordot(x1: array, x2: array, /, *, axes: Union[int, Tuple[Sequence[int],
     x1: array
         first input array. Should have a numeric data type.
     x2: array
-        second input array. Must be compatible with ``x1`` for all non-contracted axes (see :ref:`broadcasting`). Should have a numeric data type.
+        second input array. Should have a numeric data type. Corresponding contracted axes of ``x1`` and ``x2`` must be equal.
 
         .. note::
            Contracted axes (dimensions) must not be broadcasted.
