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

minor fixes (#631)

Browse Source 
* minor fixes

* var with ddof >= nelements
This commit is contained in:
Awni Hannun
2024-02-05 13:27:49 -08:00
committed by GitHub
parent d75ae52ecd
commit d40a04f8dc
5 changed files with 40 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
python/tests/test_ops.py
View File

@@ -1,4 +1,4 @@
# Copyright © 2023 Apple Inc.
# Copyright © 2023-2024 Apple Inc.
import math
import unittest
@@ -690,6 +690,14 @@ class TestOps(mlx_tests.MLXTestCase):
self.assertEqual(mx.var(x, axis=0).tolist(), [1.0, 1.0])
self.assertEqual(mx.var(x, axis=1).tolist(), [0.25, 0.25])
x = mx.array([1.0, 2.0])
out = mx.var(x, ddof=2)
self.assertEqual(out.item(), float("inf"))
x = mx.array([1.0, 2.0])
out = mx.var(x, ddof=3)
self.assertEqual(out.item(), float("inf"))
def test_abs(self):
a = mx.array([-1.0, 1.0, -2.0, 3.0])
result = mx.abs(a)