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

Adds radians and degrees (#1011)

Browse Source
This commit is contained in:
Aneesh Shetty
2024-04-22 13:17:49 -05:00
committed by GitHub
parent 3d405fb3b1
commit d0dbfe0b97
7 changed files with 120 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
python/tests/test_ops.py
View File

@@ -893,6 +893,22 @@ class TestOps(mlx_tests.MLXTestCase):
self.assertTrue(np.allclose(result, expected))
def test_degrees(self):
a = mx.array(
[0, math.pi / 4, math.pi / 2, math.pi, 3 * math.pi / 4, 2 * math.pi]
)
result = mx.degrees(a)
expected = np.degrees(a, dtype=np.float32)
self.assertTrue(np.allclose(result, expected))
def test_radians(self):
a = mx.array([0.0, 45.0, 90.0, 180.0, 270.0, 360.0])
result = mx.radians(a)
expected = np.radians(a, dtype=np.float32)
self.assertTrue(np.allclose(result, expected))
def test_log1p(self):
a = mx.array([1, 0.5, 10, 100])
result = mx.log1p(a)