tf.compat.v1.space_to_batch