@@ -90,8 +90,9 @@ def tensordot(x1, x2, axes=2):
90
90
to `x2`. Both sequences must have equal length, and each axis
91
91
`x1_axes[i]` for `x1` must have the same size as the respective
92
92
axis `x2_axes[i]` for `x2`. Each sequence must consist of unique
93
- non-negative integers that specify valid axes for each respective
94
- array.
93
+ integers that specify valid axes for each respective array.
94
+ For example, if `x1` has rank `N`, a valid axis must reside on the
95
+ half-open interval `[-N, N)`.
95
96
Returns:
96
97
usm_ndarray:
97
98
an array containing the tensor contraction whose shape consists of
@@ -310,12 +311,11 @@ def vecdot(x1, x2, axis=-1):
310
311
axis. Input arrays should be of numeric type.
311
312
axis (Optional[int]):
312
313
axis over which to compute the dot product. The axis must
313
- be an integer on the interval `[-N, N)`, where `N` is the
314
- array rank of input arrays after broadcasting rules are
315
- applied. If specified as a negative integer, the axis along
316
- which dot product is performed is counted backward from
317
- the last axes (that is `-1` refers to the last axis). By
318
- default, dot product is computed over the last axis.
314
+ be an integer on the interval `[-N, -1]`, where `N` is
315
+ ``min(x1.ndim, x2.ndim)``. The axis along which dot product
316
+ is performed is counted backward from the last axes
317
+ (that is, `-1` refers to the last axis). By default,
318
+ dot product is computed over the last axis.
319
319
Default: `-1`.
320
320
321
321
Returns:
0 commit comments