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对应的内容
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对应的内容