* fix - python interface

apache · Apr 11, 2020 · 960c47e · 960c47e
1 parent 7a619d3
commit 960c47e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/python/mxnet/numpy/linalg.py b/python/mxnet/numpy/linalg.py
@@ -22,7 +22,7 @@
 from . import fallback_linalg
 
 __all__ = ['norm', 'svd', 'cholesky', 'qr', 'inv', 'det', 'slogdet', 'solve', 'tensorinv', 'tensorsolve',
-           'pinv', 'eigvals', 'eig', 'eigvalsh', 'eigh', 'lstsq']
+           'pinv', 'eigvals', 'eig', 'eigvalsh', 'eigh', 'lstsq', 'matrix_rank']
 __all__ += fallback_linalg.__all__
 
 

diff --git a/python/mxnet/symbol/numpy/linalg.py b/python/mxnet/symbol/numpy/linalg.py
@@ -34,9 +34,9 @@ def matrix_rank(M, tol=None, hermitian=False):
     greater than `tol`.
 
     Parameters
-    M : {(M,), (..., M, N)} symbol
+    M : {(M,), (..., M, N)} _Symbol
         Input vector or stack of matrices.
-    tol : (...) symbol, float, optional
+    tol : (...) _Symbol, float, optional
         Threshold below which SVD values are considered zero. If `tol` is
         None, and ``S`` is an array with singular values for `M`, and
         ``eps`` is the epsilon value for datatype of ``S``, then `tol` is
@@ -48,7 +48,7 @@ def matrix_rank(M, tol=None, hermitian=False):
 
     Returns
     -------
-    rank : (...) symbol
+    rank : (...) _Symbol
         Rank of M.
     """
     finfo_eps_32 = _np.finfo(_np.float32).eps