case

paddle.static.nn. case ( pred_fn_pairs, default=None, name=None ) [源代码]

api_attr

声明式编程模式(静态图)

该OP的运行方式类似于python的if-elif-elif-else。

参数:

  • pred_fn_pairs (list|tuple) - 一个list或者tuple,元素是二元组(pred, fn)。其中 pred 是形状为[1]的布尔型 Tensor,fn 是一个可调用对象。所有的可调用对象都返回相同结构的Tensor。

  • default (callable,可选) - 可调用对象,返回一个或多个张量。

  • name (str,可选) – 具体用法请参见 Name ,一般无需设置,默认值:None。

返回:如果 pred_fn_pairs 中存在pred是True的元组(pred, fn),则返回第一个为True的pred的元组中fn的返回结果;如果 pred_fn_pairs 中不存在pred为True的元组(pred, fn) 且 default 不是None,则返回调用 default 的返回结果; 如果 pred_fn_pairs 中不存在pred为True的元组(pred, fn) 且 default 是None,则返回 pred_fn_pairs 中最后一个pred的返回结果。

返回类型:Tensor|list(Tensor)

抛出异常:

  • TypeError - 如果 pred_fn_pairs 的类型不是list或tuple。

  • TypeError - 如果 pred_fn_pairs 的元素的类型不是tuple。

  • TypeError - 如果 pred_fn_pairs 的tuple类型的元素大小不是2。

  • TypeError - 如果 pred_fn_pairs 中的2-tuple的第一个元素的类型不是Tensor。

  • TypeError - 如果 pred_fn_pairs 中的2-tuple的第二个元素不是可调用对象。

  • TypeError - 当 default 不是None又不是可调用对象时。

代码示例

  1. import paddle
  2. paddle.enable_static()
  3. def fn_1():
  4. return paddle.full(shape=[1, 2], dtype='float32', fill_value=1)
  5. def fn_2():
  6. return paddle.full(shape=[2, 2], dtype='int32', fill_value=2)
  7. def fn_3():
  8. return paddle.full(shape=[3], dtype='int32', fill_value=3)
  9. main_program = paddle.static.default_startup_program()
  10. startup_program = paddle.static.default_main_program()
  11. with paddle.static.program_guard(main_program, startup_program):
  12. x = paddle.full(shape=[1], dtype='float32', fill_value=0.3)
  13. y = paddle.full(shape=[1], dtype='float32', fill_value=0.1)
  14. z = paddle.full(shape=[1], dtype='float32', fill_value=0.2)
  15. pred_1 = paddle.less_than(z, x) # true: 0.2 < 0.3
  16. pred_2 = paddle.less_than(x, y) # false: 0.3 < 0.1
  17. pred_3 = paddle.equal(x, y) # false: 0.3 == 0.1
  18. # Call fn_1 because pred_1 is True
  19. out_1 = paddle.static.nn.case(
  20. pred_fn_pairs=[(pred_1, fn_1), (pred_2, fn_2)], default=fn_3)
  21. # Argument default is None and no pred in pred_fn_pairs is True. fn_3 will be called.
  22. # because fn_3 is the last callable in pred_fn_pairs.
  23. out_2 = paddle.static.nn.case(pred_fn_pairs=[(pred_2, fn_2), (pred_3, fn_3)])
  24. exe = paddle.static.Executor(paddle.CPUPlace())
  25. res_1, res_2 = exe.run(main_program, fetch_list=[out_1, out_2])
  26. print(res_1) # [[1. 1.]]
  27. print(res_2) # [3 3 3]