cupy.linalg.solve¶
-
cupy.linalg.
solve
(a, b)[source]¶ Solves a linear matrix equation.
It computes the exact solution of
x
inax = b
, wherea
is a square and full rank matrix.Parameters: - a (cupy.ndarray) – The matrix with dimension
(M, M)
- b (cupy.ndarray) – The vector with
M
elements, or the matrix with dimension(M, K)
Returns: The vector with
M
elements, or the matrix with dimension(M, K)
.Return type: See also
- a (cupy.ndarray) – The matrix with dimension