_scimath_impl.pyi 2.7 KB

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