scimath.pyi 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  1. from typing import overload, Any
  2. from numpy import complexfloating
  3. from numpy._typing import (
  4. NDArray,
  5. _ArrayLikeFloat_co,
  6. _ArrayLikeComplex_co,
  7. _ComplexLike_co,
  8. _FloatLike_co,
  9. )
  10. __all__: list[str]
  11. @overload
  12. def sqrt(x: _FloatLike_co) -> Any: ...
  13. @overload
  14. def sqrt(x: _ComplexLike_co) -> complexfloating[Any, Any]: ...
  15. @overload
  16. def sqrt(x: _ArrayLikeFloat_co) -> NDArray[Any]: ...
  17. @overload
  18. def sqrt(x: _ArrayLikeComplex_co) -> NDArray[complexfloating[Any, Any]]: ...
  19. @overload
  20. def log(x: _FloatLike_co) -> Any: ...
  21. @overload
  22. def log(x: _ComplexLike_co) -> complexfloating[Any, Any]: ...
  23. @overload
  24. def log(x: _ArrayLikeFloat_co) -> NDArray[Any]: ...
  25. @overload
  26. def log(x: _ArrayLikeComplex_co) -> NDArray[complexfloating[Any, Any]]: ...
  27. @overload
  28. def log10(x: _FloatLike_co) -> Any: ...
  29. @overload
  30. def log10(x: _ComplexLike_co) -> complexfloating[Any, Any]: ...
  31. @overload
  32. def log10(x: _ArrayLikeFloat_co) -> NDArray[Any]: ...
  33. @overload
  34. def log10(x: _ArrayLikeComplex_co) -> NDArray[complexfloating[Any, Any]]: ...
  35. @overload
  36. def log2(x: _FloatLike_co) -> Any: ...
  37. @overload
  38. def log2(x: _ComplexLike_co) -> complexfloating[Any, Any]: ...
  39. @overload
  40. def log2(x: _ArrayLikeFloat_co) -> NDArray[Any]: ...
  41. @overload
  42. def log2(x: _ArrayLikeComplex_co) -> NDArray[complexfloating[Any, Any]]: ...
  43. @overload
  44. def logn(n: _FloatLike_co, x: _FloatLike_co) -> Any: ...
  45. @overload
  46. def logn(n: _ComplexLike_co, x: _ComplexLike_co) -> complexfloating[Any, Any]: ...
  47. @overload
  48. def logn(n: _ArrayLikeFloat_co, x: _ArrayLikeFloat_co) -> NDArray[Any]: ...
  49. @overload
  50. def logn(n: _ArrayLikeComplex_co, x: _ArrayLikeComplex_co) -> NDArray[complexfloating[Any, Any]]: ...
  51. @overload
  52. def power(x: _FloatLike_co, p: _FloatLike_co) -> Any: ...
  53. @overload
  54. def power(x: _ComplexLike_co, p: _ComplexLike_co) -> complexfloating[Any, Any]: ...
  55. @overload
  56. def power(x: _ArrayLikeFloat_co, p: _ArrayLikeFloat_co) -> NDArray[Any]: ...
  57. @overload
  58. def power(x: _ArrayLikeComplex_co, p: _ArrayLikeComplex_co) -> NDArray[complexfloating[Any, Any]]: ...
  59. @overload
  60. def arccos(x: _FloatLike_co) -> Any: ...
  61. @overload
  62. def arccos(x: _ComplexLike_co) -> complexfloating[Any, Any]: ...
  63. @overload
  64. def arccos(x: _ArrayLikeFloat_co) -> NDArray[Any]: ...
  65. @overload
  66. def arccos(x: _ArrayLikeComplex_co) -> NDArray[complexfloating[Any, Any]]: ...
  67. @overload
  68. def arcsin(x: _FloatLike_co) -> Any: ...
  69. @overload
  70. def arcsin(x: _ComplexLike_co) -> complexfloating[Any, Any]: ...
  71. @overload
  72. def arcsin(x: _ArrayLikeFloat_co) -> NDArray[Any]: ...
  73. @overload
  74. def arcsin(x: _ArrayLikeComplex_co) -> NDArray[complexfloating[Any, Any]]: ...
  75. @overload
  76. def arctanh(x: _FloatLike_co) -> Any: ...
  77. @overload
  78. def arctanh(x: _ComplexLike_co) -> complexfloating[Any, Any]: ...
  79. @overload
  80. def arctanh(x: _ArrayLikeFloat_co) -> NDArray[Any]: ...
  81. @overload
  82. def arctanh(x: _ArrayLikeComplex_co) -> NDArray[complexfloating[Any, Any]]: ...