paddlepaddle学习笔记（2）control flow控制流之if else判断

最新推荐文章于 2024-12-24 12:02:04 发布

转载最新推荐文章于 2024-12-24 12:02:04 发布 · 567 阅读

文章标签：

#paddle控制流

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

21 篇文章

订阅专栏

python

15 篇文章

订阅专栏

paddlepaddle

8 篇文章

订阅专栏

本文通过一个具体的代码示例，展示了如何在PaddlePaddle中使用IfElse操作符来实现条件执行流程，包括定义变量、进行比较判断及执行相应代码块。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、代码

from paddle import fluid
a = fluid.layers.fill_constant(shape=[2,1],dtype='int64',value=5)
b= fluid.layers.fill_constant(shape=[2,1],dtype='int64',value=6)
ifcond = fluid.layers.less_than(x=a,y=b)#是否a<b的判断
ie = fluid.layers.IfElse(ifcond)#IfElse执行这个判断，如果正确则执行。这里a是小于b的
with ie.true_block():  #以下几行代码实际上就是把a加1，然后输出
    c = ie.input(a)
    c+=1
    ie.output(c)
exe = fluid.Executor(fluid.CPUPlace())
exe.run(fluid.default_main_program(),fetch_list=[c])

二、执行结果