轉換圖像塊的3D張量成2D圖像上TensorFlow

Question

我有分爲大小32 x 32的不重疊的塊，並排列爲大小[64, 32, 32]的3D張量大小256 x 256的圖像。這裏，64是圖像中32 x 32塊的數量。 64個塊的排列方式使得前8個塊形成第一行，下8個第二行等等。

我想知道是否有一種方法可以在不使用循環的情況下給出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