Fix docstring of `polyfit`
@@ -140,14 +140,16 @@ def polyfit(x, y, deg, rcond=None, full=False, w=None, cov=False):
along with the covariance matrix.
Returns:
- cupy.ndarray: of shape (deg + 1,) or (deg + 1, K).
+ cupy.ndarray or tuple:
+ p (cupy.ndarray of shape (deg + 1,) or (deg + 1, K)):
Polynomial coefficients from highest to lowest degree
- tuple (cupy.ndarray, int, cupy.ndarray, float):
+ residuals, rank, singular_values, rcond
+ (cupy.ndarray, int, cupy.ndarray, float):
Present only if ``full=True``.
Sum of squared residuals of the least-squares fit,
rank of the scaled Vandermonde coefficient matrix,
its singular values, and the specified value of ``rcond``.
- cupy.ndarray: of shape (M, M) or (M, M, K).
+ V (cupy.ndarray of shape (M, M) or (M, M, K)):
Present only if ``full=False`` and ``cov=True``.
The covariance matrix of the polynomial coefficient estimates.
