tensorflow tf.where截断