Skip to content

Commit 9efd1a3

Browse files
authored
Fix guidance for tensordot broadcasting (#324)
* Fix guidance for tensordot broadcasting * Update guidance * Update note
1 parent 2f3ba6c commit 9efd1a3

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

spec/API_specification/linear_algebra_functions.md

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,11 @@ Returns a tensor contraction of `x1` and `x2` over specific axes.
8787

8888
- **x2**: _<array>_
8989

90-
- second input array. Must be compatible with `x1` (see {ref}`broadcasting`). Should have a numeric data type.
90+
- second input array. Must be compatible with `x1` for all non-contracted axes (see {ref}`broadcasting`). Should have a numeric data type.
91+
92+
```{note}
93+
Contracted axes (dimensions) must not be broadcasted.
94+
```
9195
9296
- **axes**: _Union\[ int, Tuple\[ Sequence\[ int ], Sequence\[ int ] ] ]_
9397

0 commit comments

Comments
 (0)