()
| 304 | |
| 305 | |
| 306 | def test_dots(): |
| 307 | n1, n2, n3, n4 = 100, 50, 200, 100 |
| 308 | |
| 309 | rng = np.random.RandomState(0) |
| 310 | |
| 311 | A = rng.randn(n1, n2) |
| 312 | B = rng.randn(n2, n3) |
| 313 | C = rng.randn(n3, n4) |
| 314 | |
| 315 | X1 = ot.utils.dots(A, B, C) |
| 316 | |
| 317 | X2 = A.dot(B.dot(C)) |
| 318 | |
| 319 | np.testing.assert_allclose(X1, X2) |
| 320 | |
| 321 | |
| 322 | def test_clean_zeros(): |