tf.space_to_batch