Method forward

torchscript/loop_script_onnx.py:11–12

(self, x, y)

class LoopModel2(torch.nn.Module):
    def forward(self, x, y):
        return loop(x, y)

model = LoopModel2()
dummy_input = torch.ones(2, 3, dtype=torch.long)