Method main

tests/python/relax/test_op_size.py:30–31

(x: R.Tensor((2, 3), "float32"))

    class Module:
        @R.function
        def main(x: R.Tensor((2, 3), "float32")) -> R.Tensor((), "int64"):
            return R.size(x)

    x_np = np.random.rand(2, 3).astype("float32")
    x = tvm.runtime.tensor(x_np)