From 3a6857c99d9012650b8b58726783eee30308d944 Mon Sep 17 00:00:00 2001 From: Bas van Beek Date: Wed, 21 Oct 2020 13:01:11 +0200 Subject: MAINT: Update the `axes` parameter of `tensordot` `axes` can take either an integer or a 2-tuple of shape-like objects --- numpy/core/numeric.pyi | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'numpy/core/numeric.pyi') 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, -- cgit v1.2.1