12
12
13
13
from ._types import Literal , Optional , Tuple , Union , array
14
14
from .constants import inf
15
+ from collections .abc import Sequence
15
16
16
17
def cholesky (x : array , / , * , upper : bool = False ) -> array :
17
18
pass
@@ -46,7 +47,7 @@ def inv(x: array, /) -> array:
46
47
def lstsq (x1 : array , x2 : array , / , * , rtol : Optional [Union [float , array ]] = None ) -> Tuple [array , array , array , array ]:
47
48
pass
48
49
49
- def matmul (x1 , x2 , / ):
50
+ def matmul (x1 : array , x2 : array , / ) -> array :
50
51
pass
51
52
52
53
def matrix_power (x : array , n : int , / ) -> array :
@@ -76,7 +77,7 @@ def solve(x1: array, x2: array, /) -> array:
76
77
def svd (x : array , / , * , full_matrices : bool = True ) -> Union [array , Tuple [array , ...]]:
77
78
pass
78
79
79
- def tensordot (x1 , x2 , / , * , axes = 2 ) :
80
+ def tensordot (x1 : array , x2 : array , / , * , axes : Union [ int , Tuple [ Sequence [ int ], Sequence [ int ]]] = 2 ) -> array :
80
81
pass
81
82
82
83
def svdvals (x : array , / ) -> Union [array , Tuple [array , ...]]:
@@ -85,10 +86,10 @@ def svdvals(x: array, /) -> Union[array, Tuple[array, ...]]:
85
86
def trace (x : array , / , * , axis1 : int = 0 , axis2 : int = 1 , offset : int = 0 ) -> array :
86
87
pass
87
88
88
- def transpose (x , / , * , axes = None ):
89
+ def transpose (x : array , / , * , axes : Optional [ Tuple [ int , ...]] = None ) -> array :
89
90
pass
90
91
91
- def vecdot (x1 , x2 , / , * , axis = None ):
92
+ def vecdot (x1 : array , x2 : array , / , * , axis : Optional [ int ] = None ) -> array :
92
93
pass
93
94
94
95
__all__ = ['cholesky' , 'cross' , 'det' , 'diagonal' , 'eig' , 'eigh' , 'eigvals' , 'eigvalsh' , 'einsum' , 'inv' , 'lstsq' , 'matmul' , 'matrix_power' , 'matrix_rank' , 'norm' , 'outer' , 'pinv' , 'qr' , 'slogdet' , 'solve' , 'svd' , 'tensordot' , 'svdvals' , 'trace' , 'transpose' , 'vecdot' ]
0 commit comments