功能介绍
根据条件返回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”。
约束说明