Function fun

imperative/python/test/unit/jit/test_tracing.py:172–173

(data, repeats)

def test_repeat_in_trace():
    @trace(symbolic=False)
    def fun(data, repeats):
        F.repeat(data, repeats)

    data = tensor(np.random.random([1, 2, 3]).astype(np.float32))