torch.ops.script_ops.while_loop这个怎么用

torch.ops.script_ops.while_loop是PyTorch中的一个函数，用于在脚本模式下执行循环。它接受三个参数：
1. cond: 循环条件，是一个函数，每次迭代时调用并返回一个布尔值。当返回值为True时继续循环，否则退出循环。
2. body: 循环体，是一个函数，每次迭代时调用。
3. loop_vars: 循环变量，是一个元组，代表循环中需要更新的变量。

例如：

import torch

def cond(i):
    return i < 3

def body(i):
    print(i)
    return (i + 1,)

i = torch.tensor(0, dtype=torch.int32)
torch.ops.script_ops.while_loop(cond, body, (i,))

这个例子中，循环条件为i < 3, 循环体是打印出i, 循环变量是i, 结果是打印出0,1,2