Function run

imperative/python/test/unit/functional/test_tensor.py:54–55

(data)

    for axis in range(-2, 3):

        def run(data):
            return F.diag(data, k=axis)

        opr_test(cases, run, ref_fn=lambda x: np.diag(x, axis), network=network)