_umath_linalg.pyi 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. from typing import Final, Literal as L
  2. import numpy as np
  3. from numpy._typing._ufunc import _GUFunc_Nin2_Nout1
  4. __version__: Final[str] = ...
  5. _ilp64: Final[bool] = ...
  6. ###
  7. # 1 -> 1
  8. # (m,m) -> ()
  9. det: Final[np.ufunc] = ...
  10. # (m,m) -> (m)
  11. cholesky_lo: Final[np.ufunc] = ...
  12. cholesky_up: Final[np.ufunc] = ...
  13. eigvals: Final[np.ufunc] = ...
  14. eigvalsh_lo: Final[np.ufunc] = ...
  15. eigvalsh_up: Final[np.ufunc] = ...
  16. # (m,m) -> (m,m)
  17. inv: Final[np.ufunc] = ...
  18. # (m,n) -> (p)
  19. qr_r_raw: Final[np.ufunc] = ...
  20. svd: Final[np.ufunc] = ...
  21. ###
  22. # 1 -> 2
  23. # (m,m) -> (), ()
  24. slogdet: Final[np.ufunc] = ...
  25. # (m,m) -> (m), (m,m)
  26. eig: Final[np.ufunc] = ...
  27. eigh_lo: Final[np.ufunc] = ...
  28. eigh_up: Final[np.ufunc] = ...
  29. ###
  30. # 2 -> 1
  31. # (m,n), (n) -> (m,m)
  32. qr_complete: Final[_GUFunc_Nin2_Nout1[L["qr_complete"], L[2], None, L["(m,n),(n)->(m,m)"]]] = ...
  33. # (m,n), (k) -> (m,k)
  34. qr_reduced: Final[_GUFunc_Nin2_Nout1[L["qr_reduced"], L[2], None, L["(m,n),(k)->(m,k)"]]] = ...
  35. # (m,m), (m,n) -> (m,n)
  36. solve: Final[_GUFunc_Nin2_Nout1[L["solve"], L[4], None, L["(m,m),(m,n)->(m,n)"]]] = ...
  37. # (m,m), (m) -> (m)
  38. solve1: Final[_GUFunc_Nin2_Nout1[L["solve1"], L[4], None, L["(m,m),(m)->(m)"]]] = ...
  39. ###
  40. # 1 -> 3
  41. # (m,n) -> (m,m), (p), (n,n)
  42. svd_f: Final[np.ufunc] = ...
  43. # (m,n) -> (m,p), (p), (p,n)
  44. svd_s: Final[np.ufunc] = ...
  45. ###
  46. # 3 -> 4
  47. # (m,n), (m,k), () -> (n,k), (k), (), (p)
  48. lstsq: Final[np.ufunc] = ...