fix doc string

This commit is contained in:
zhujingxuan 2021-12-16 20:01:55 +08:00
parent f85bb6f5ea
commit 60c64eb954
2 changed files with 27 additions and 23 deletions

View File

@ -116,8 +116,8 @@ def solve_triangular(A, b, trans=0, lower=False, unit_diagonal=False,
Returns:
Tensor of shape :math:`(M,)` or :math:`(M, N)`,
which is the solution to the system :math:`A x = b`.
Shape of :math:`x` matches :math:`b`.
which is the solution to the system :math:`A x = b`.
Shape of :math:`x` matches :math:`b`.
Raises:
LinAlgError: If :math:`A` is singular
@ -213,9 +213,9 @@ def cho_factor(a, lower=False, overwrite_a=False, check_finite=True):
(crashes, non-termination) if the inputs do contain infinities or NaNs.
Returns:
Tensor, Matrix whose upper or lower triangle contains the Cholesky factor of `a`.
Other parts of the matrix contain random data.
bool, Flag indicating whether the factor is in the lower or upper triangle
- Tensor, Matrix whose upper or lower triangle contains the Cholesky factor of `a`.
Other parts of the matrix contain random data.
- bool, Flag indicating whether the factor is in the lower or upper triangle
Raises:
LinAlgError: Raised if decomposition fails.
@ -357,9 +357,9 @@ def eigh(a, b=None, lower=True, eigvals_only=False, overwrite_a=False,
and eigenvectors are returned.
Returns:
Tensor with shape (N,), The N (1<=N<=M) selected eigenvalues, in ascending order,
each repeated according to its multiplicity.
Tensor with shape (M, N), (if ``eigvals_only == False``)
- Tensor with shape (N,), The N (1<=N<=M) selected eigenvalues, in ascending order,
each repeated according to its multiplicity.
- Tensor with shape (M, N), (if ``eigvals_only == False``)
Raises:
LinAlgError: If eigenvalue computation does not converge, an error occurred, or b matrix is not
@ -457,9 +457,9 @@ def lu_factor(a, overwrite_a=False, check_finite=True):
Returns:
Tensor, a square matrix of (N, N) containing U in its upper triangle, and L in its lower triangle.
The unit diagonal elements of L are not stored.
The unit diagonal elements of L are not stored.
Tensor, (N,) Pivot indices representing the permutation matrix P:
row i of matrix was interchanged with row piv[i].
row i of matrix was interchanged with row piv[i].
Supported Platforms:
``CPU`` ``GPU``
@ -506,16 +506,14 @@ def lu(a, permute_l=False, overwrite_a=False, check_finite=True):
Returns:
**(If permute_l == False)**
Tensor, (M, M) Permutation matrix
Tensor, (M, K) Lower triangular or trapezoidal matrix with unit diagonal.
K = min(M, N)
Tensor, (K, N) Upper triangular or trapezoidal matrix
- Tensor, (M, M) Permutation matrix
- Tensor, (M, K) Lower triangular or trapezoidal matrix with unit diagonal. K = min(M, N)
- Tensor, (K, N) Upper triangular or trapezoidal matrix
**(If permute_l == True)**
Tensor, (M, K) Permuted L matrix.
K = min(M, N)
Tensor, (K, N) Upper triangular or trapezoidal matrix
- Tensor, (M, K) Permuted L matrix. K = min(M, N)
- Tensor, (K, N) Upper triangular or trapezoidal matrix
Supported Platforms:
``CPU`` ``GPU``

View File

@ -187,6 +187,10 @@ def gmres(A, b, x0=None, *, tol=1e-5, atol=0.0, restart=20, maxiter=None,
need not have any particular special properties, such as symmetry. However,
convergence is often slow for nearly symmetric operators.
Note:
In the future, MindSpore will report the number of iterations when convergence
is not achieved, like SciPy. Currently it is None, as a Placeholder.
Args:
A (Tensor or function): 2D Tensor or function that calculates the linear
map (matrix-vector product) ``Ax`` when called like ``A(x)``.
@ -225,9 +229,8 @@ def gmres(A, b, x0=None, *, tol=1e-5, atol=0.0, restart=20, maxiter=None,
early termination, but has much less overhead on GPUs.
Returns:
Tensor, The converged solution. Has the same structure as ``b``.
None, Placeholder for convergence information. In the future, MindSpore
will report the number of iterations when convergence is not achieved, like SciPy.
- Tensor, The converged solution. Has the same structure as ``b``.
- None, Placeholder for convergence information.
Supported Platforms:
``CPU`` ``GPU``
@ -324,6 +327,10 @@ def cg(A, b, x0=None, *, tol=1e-5, atol=0.0, maxiter=None, M=None):
another ``cg`` solve, rather than by differentiating *through* the solver.
They will be accurate only if both solves converge.
Note:
In the future, MindSpore will report the number of iterations when convergence
is not achieved, like SciPy. Currently it is None, as a Placeholder.
Args:
A (Tensor or function): 2D Tensor or function that calculates the linear
map (matrix-vector product) ``Ax`` when called like ``A(x)``.
@ -342,9 +349,8 @@ def cg(A, b, x0=None, *, tol=1e-5, atol=0.0, maxiter=None, M=None):
to reach a given error tolerance.
Returns:
Tensor, The converged solution. Has the same structure as ``b``.
None, Placeholder for convergence information. In the future, MindSpore will report
the number of iterations when convergence is not achieved, like SciPy.
- Tensor, The converged solution. Has the same structure as ``b``.
- None, Placeholder for convergence information.
Supported Platforms:
``CPU`` ``GPU``