(self)
| 29 | self.assertFalse(mx.array_equal(k1, k2)) |
| 30 | |
| 31 | def test_key_split(self): |
| 32 | key = mx.random.key(0) |
| 33 | |
| 34 | k1, k2 = mx.random.split(key) |
| 35 | self.assertFalse(mx.array_equal(k1, k2)) |
| 36 | |
| 37 | r1, r2 = mx.random.split(key) |
| 38 | self.assertTrue(mx.array_equal(k1, r1)) |
| 39 | self.assertTrue(mx.array_equal(k2, r2)) |
| 40 | |
| 41 | keys = mx.random.split(key, 10) |
| 42 | self.assertEqual(keys.shape, (10, 2)) |
| 43 | |
| 44 | def test_uniform(self): |
| 45 | key = mx.random.key(0) |