テンソルフローのマージディメンションと逆マージ

Question

テンソルの形状が[3, 5, 200, 300, 3]であるとします。どのようにして最初の2つのdimを一緒にマージして、テンソルの形が[15, 200, 300, 3]になるようにすることができますか？その後、私はマージ操作を逆にして元の形にすることができます。

Answer 1

あなただけ使用することができますtf.reshape

a = tf.random_normal(shape=(3, 5, 200, 300, 3)) 
b = tf.reshape(a, shape=(15, 200, 300, 3)) 
... 
c = tf.reshape(b, shape=(3, 5, 200, 300, 3))