diff options
| author | Bas van Beek <b.f.van.beek@vu.nl> | 2020-10-21 13:01:11 +0200 |
|---|---|---|
| committer | Bas van Beek <b.f.van.beek@vu.nl> | 2020-10-21 13:01:11 +0200 |
| commit | 3a6857c99d9012650b8b58726783eee30308d944 (patch) | |
| tree | 829e8eca96715106ac2c9acfd0fe63708edeac0a /numpy/core/numeric.pyi | |
| parent | 71292e07e0329d0b256804146d06b99d83fd36c9 (diff) | |
| download | numpy-3a6857c99d9012650b8b58726783eee30308d944.tar.gz | |
MAINT: Update the `axes` parameter of `tensordot`
`axes` can take either an integer or a 2-tuple of shape-like objects
Diffstat (limited to 'numpy/core/numeric.pyi')
| -rw-r--r-- | numpy/core/numeric.pyi | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/numpy/core/numeric.pyi b/numpy/core/numeric.pyi index b341d136a..f917f74de 100644 --- a/numpy/core/numeric.pyi +++ b/numpy/core/numeric.pyi @@ -126,7 +126,7 @@ def outer(a: ArrayLike, b: ArrayLike, out: _ArrayType = ...) -> _ArrayType: ... def tensordot( a: ArrayLike, b: ArrayLike, - axes: _ShapeLike = ..., + axes: Union[int, Tuple[_ShapeLike, _ShapeLike]] = ..., ) -> ndarray: ... def roll( a: ArrayLike, |