tf.nest.pack_sequence_as