请提出你的问题 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])
1条答案
按热度按时间cotxawn71#
你好,请参考这个文档:https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/static/nn/while_loop_cn.html#while-loop
我在python38 paddle2.0的环境下测试了,输出正常。