zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-10 04:20:09 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added ArcTan2 operation (#1079)

Browse Source 
* Added ArcTan2 operation

* Cleanup, bug fixes from code review

* Minor cleanup, fixed Linux tests
This commit is contained in:
Rahul Yedida
2024-05-08 11:35:15 -04:00
committed by GitHub
parent fe96ceee66
commit cc05a281c4
16 changed files with 143 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/tests/test_ops.py
View File

@@ -1273,6 +1273,7 @@ class TestOps(mlx_tests.MLXTestCase):
"arcsin": lambda primal, cotan: cotan / np.sqrt(1.0 - primal**2),
"arccos": lambda primal, cotan: -cotan / np.sqrt(1.0 - primal**2),
"arctan": lambda primal, cotan: cotan / (1.0 + primal**2),
"arctan2": lambda primal, cotan: cotan / (1.0 + primal**2),
"arcsinh": lambda primal, cotan: cotan / np.sqrt(primal**2 + 1),
"arccosh": lambda primal, cotan: cotan / np.sqrt(primal**2 - 1),
"arctanh": lambda primal, cotan: cotan / (1.0 - primal**2),