tf.depth_to_space详解

tf.depth_to_space ( 
    input , 
    block_size , 
    name = None
 )

将数据从深度重新排列为空间数据块.这是 SpaceToDepth 的逆向转换.更具体地说,此 op 输出输入张量的一个副本,其中来自深度维度的值在空间块中移动到高度和宽度维度.attr block_size 表示输入块的大小以及数据的移动方式.

从深度大小为 block_size * block_size 的数据块重新排列成不重叠的大小为 block_size x block_size 的数据块.
输出张量的宽度为 input_depth * block_size,而高度是 input_height * block_size.

输入张量的深度必须可以被 block_size * block_size 整除.

即,假设输入的形状是:[batch, height, width, depth],输出的形状为:[batch, heightblock_size, widthblock_size, depth/(block_size*block_size)]；