Function loop

torchscript/loop_script_onnx.py:5–8

(x, y)

@torch.jit.script
def loop(x, y):
    for i in range(int(y)):
        x = x + i
    return x

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