(self)
| 876 | self.assertEqual(y.shape, (4, 7, 7, 8)) |
| 877 | |
| 878 | def test_sequential(self): |
| 879 | x = mx.ones((10, 2)) |
| 880 | m = nn.Sequential(nn.Linear(2, 10), nn.ReLU(), nn.Linear(10, 1)) |
| 881 | y = m(x) |
| 882 | self.assertEqual(y.shape, (10, 1)) |
| 883 | params = m.parameters() |
| 884 | self.assertTrue("layers" in params) |
| 885 | self.assertEqual(len(params["layers"]), 3) |
| 886 | self.assertTrue("weight" in params["layers"][0]) |
| 887 | self.assertEqual(len(params["layers"][1]), 0) |
| 888 | self.assertTrue("weight" in params["layers"][2]) |
| 889 | |
| 890 | m.layers[1] = nn.relu |
| 891 | y2 = m(x) |
| 892 | self.assertTrue(mx.array_equal(y, y2)) |
| 893 | |
| 894 | def test_gelu(self): |
| 895 | inputs = [1.15286231, -0.81037411, 0.35816911, 0.77484438, 0.66276414] |
nothing calls this directly
no test coverage detected