tf.nn.depth_to_space