把tf.map_fn替换成tf.where实现

举例：
把
label_tensor = tf.map_fn(lambda x: tf.cond(x >= 1, lambda: 1, lambda: 0), label_tensor) 改成
label_tensor = tf.where(label_tensor >= 1, tf.ones_like(label_tensor), tf.zeros_like(label_tensor))