Function cb

imperative/python/test/unit/random/test_rng.py:715–716

(grad)

    g = []

    def cb(grad):
        g.append(grad)

    n, m = 6, 3
    arr = np.arange(n * m)