tensorflow tf.where使用方法，大于某个值为1,小于为0

one = tf.ones_like(label)

zero = tf.zeros_like(label)

label = tf.where(label <0.5, x=zero, y=one)

one是全为1的，zero是全为0。

tf.where的作用是第一项为true时，label等于x对应的内容；为false时，label等于y对应的内容