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

c90pui9n  于 5个月前  发布在  其他
关注(0)|答案(1)|浏览(44)

请提出你的问题 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])

cotxawn7

cotxawn71#

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

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

相关问题