where

功能介绍

根据条件返回x或y中的元素。

函数原型

1
2
3
where(
    condition, x, y, name=None
)

用法

用法同tf原生的where算子,condition为bool类型,x和y为特定类型入参(tf.int64)会执行SVE优化后的算子;其他类型入参则执行原生的where算子。

具体安装方法可参考Rec SDK源码中的“cust_op/tf_cpu_op/README.md”

约束说明

  • 仅支持1维Tensor入参。
  • x与y入参类型一致。
  • condition、x和y的长度需要一致。