tf.where()

https://blog.csdn.net/A_a_ron/article/details/79048446