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

最新推荐文章于 2025-12-23 19:06:22 发布

转载最新推荐文章于 2025-12-23 19:06:22 发布 · 480 阅读

文章标签：

#paddle控制流 #paddle的while循环

深度学习同时被 3 个专栏收录

21 篇文章

订阅专栏

机器学习

18 篇文章

订阅专栏

python

15 篇文章

订阅专栏

本文通过一个具体的代码示例，展示了如何在PaddlePaddle中使用循环结构进行数组操作，包括初始化、写入、读取和更新数组元素，并在CPU环境下执行程序。

一、代码

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])

二、运行结果