Method test

tests/python/relax/test_frontend_nn_op.py:36–39

(self, x: Tensor)

def test_unary():
    class Model(Module):
        def test(self, x: Tensor):
            z0 = op.square(x)
            z1 = op.sqrt(x)
            return (z0, z1)

    # fmt: off
    @R.function