Allow burn-in in memory leak test

tests/tensor/test_elemwise.py

@@ -242,7 +242,8 @@ class TestDimShuffle(unittest_tools.InferShapeTester):
             blocks_last = blocks_i
 
         tracemalloc.stop()
-        assert np.allclose(np.mean(block_diffs), 0)
+        burn_in = 1
+        assert np.allclose(np.mean(block_diffs[burn_in:]), 0)
 
     def test_static_shape(self):
         x = tensor(dtype=np.float64, shape=(1, 2), name="x")