tf.nest.flatten