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)]；