_umath_linalg.pyi 1.4 KB

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