paddlepaddle2.0静态图模型实现循环

[VVJY]

请提出你的问题 Please ask your question

paddle2.0版本静态图模式下，想要对一个batch的ins做循环，以下各种用法都提示不存在接口，请问正确的用法应该是怎样的，或者怎样高效实现循环呢：
[i] = paddle.static.While(loop_cond, loop_body,[i])
报错：'module' object has no attribute 'While'
[i] = paddle.static.nn.control_flow.While(loop_cond, loop_body,[i])
报错：'module' object has no attribute 'control_flow'
[i] = paddle.static.nn.while_loop(loop_cond, loop_body,[i])

[minghaoBD]

你好，请参考这个文档：https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/static/nn/while_loop_cn.html#while-loop

我在python38 paddle2.0的环境下测试了，输出正常。