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

make python array SupportsAbs conform (like numpy) (#624)

Browse Source
This commit is contained in:
Daniel Strobusch
2024-02-04 18:31:02 +01:00
committed by GitHub
parent 9852af1a19
commit 4fd2fb84a6
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_ops.py
View File

@@ -696,6 +696,8 @@ class TestOps(mlx_tests.MLXTestCase):
expected = np.abs(a, dtype=np.float32)
self.assertTrue(np.allclose(result, expected))
self.assertTrue(np.allclose(a.abs(), abs(a)))
def test_negative(self):
a = mx.array([-1.0, 1.0, -2.0, 3.0])
result = mx.negative(a)