where 功能介绍 根据条件返回x或y中的元素。 函数原型 1 2 3where( 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的长度需要一致。 父主题: Host算子