Switch

    该类用于实现Switch分支控制功能。Switch分支包含多个case分支和一个default分支，Switch控制流会依次检查各case分支条件是否满足，并仅执行第一个满足条件的case分支后面的语句。若不存在满足条件的case分支，则仅执行default分支后面的语句。

    注解

    成员函数：

    • case(cond) - Switch的case分支，其参数cond为bool型的标量Variable。只有当前case分支的cond为True，且之前的case分支的cond均为False，该case分支后的语句才会执行，且不再执行之后的case后的语句。

    注意：case和default函数只能用于Switch的scope内部，示例如下：

    参数：

    • name (str，可选) - 具体用法请参见 ，一般无需设置，默认值为None。
    1. import paddle.fluid as fluid
    3. lr = fluid.layers.create_global_var(
    4.     shape=[1],
    5.     value=0.0,
    6.     dtype='float32',
    7.     persistable=True,
    8. zero_var = fluid.layers.fill_constant(
    9.     shape=[1], dtype='float32', value=0.0)
    10. one_var = fluid.layers.fill_constant(
    11.     shape=[1], dtype='float32', value=1.0)
    12. two_var = fluid.layers.fill_constant(
    13.     shape=[1], dtype='float32', value=2.0)
    15. # 将参数中的begin设为非0值，则进入Switch的default分支，输出数组中的数字将为2
    16. global_step = fluid.layers.autoincreased_step_counter(counter_name='@LR_DECAY_COUNTER@', begin=0, step=1)
    18.     with switch.case(global_step == zero_var):
    19.         fluid.layers.assign(input=one_var, output=lr)
    20.     with switch.default():
    21.         fluid.layers.assign(input=two_var, output=lr)
    23. exe = fluid.Executor(fluid.CPUPlace())
    24. exe.run(fluid.default_startup_program())
    26. res = exe.run(fluid.default_main_program(), feed={}, fetch_list=[lr])
    27. print(res) # [array([1.], dtype=float32)]