paddlepaddle学习笔记（3）control flow控制流之while循环

一、代码

from paddle import fluid
a = fluid.layers.fill_constant(shape=[2,2],dtype="float32",value=1.0)
i = fluid.layers.zeros([1],dtype="int64")
until = fluid.layers.fill_constant([1],dtype="int64",value=10)
data_arr = fluid.layers.array_write(a,i)
cond = fluid.layers.less_than(x=i,y=until)
while_op = fluid.layers.While(cond=cond)
with while_op.block():
    a = fluid.layers.array_read(data_arr,i)
    a = a+1
    i = fluid.layers.increment(x=i,value=1,in_place=True)
    fluid.layers.less_than(x=i,y=until,cond=cond)
    fluid.layers.array_write(a,i,data_arr)
ret = fluid.layers.array_read(data_arr,i-1)
exe = fluid.Executor(fluid.CPUPlace())
exe.run(fluid.default_startup_program())
exe.run(fetch_list=[ret])
    
    

二、运行结果