用法一:
tf.where(bool, a, b)
#bool,a, b 都为tensor张量,bool为真返回a,为假返回b用法二:
tf.where(tensor, a, b)
#a,b和tensor为相同维度的tensor张量,将tensor中的true位置元素替换为a中对应位置元素,false的替换为b中对应位置元素。
版权声明:本文为goodman_lqifei原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
用法一:
tf.where(bool, a, b)
#bool,a, b 都为tensor张量,bool为真返回a,为假返回b用法二:
tf.where(tensor, a, b)
#a,b和tensor为相同维度的tensor张量,将tensor中的true位置元素替换为a中对应位置元素,false的替换为b中对应位置元素。