TensorFlow上で3Dテンソルの画像ブロックを2D画像に変換する

Question

私は、サイズ256 x 256の画像を、サイズが32 x 32の非重複ブロックに分割し、サイズ[64, 32, 32]の3Dテンソルとして配置しました。ここで、64は、画像内の32 x 32ブロックの数です。 64ブロックは、最初の8ブロックが第1の行を形成し、次の8ブロックが第2の行を形成するように配列される。

ループを使用せずにTensorFlowで画像ブロックを指定して完全な画像を作成する方法があるかどうかを知りたいと思います。関連する関数tf.batch_to_spaceがありますが、必要なことを正確に実行するわけではありません。助けてください。

Answer 1

def reconstruct(x): 
    x = tf.reshape(x, [8, 8, 32, 32]) 
    x = tf.transpose(x, [0, 2, 1, 3]) # x.shape is [8, 32, 8, 32] 
    x = tf.reshape(x, [256, 256])  # Because tf tensor is row-major. 
    return x