在做多張的圖片為單個輸入的batch的時候,調用預訓練網絡,需要先reshape到網絡結構,
然后再reshape回原來的形狀,這里需要兩次reshape之后能夠保持原有的狀態(tài):
以下為對reshape的測試:
import tensorflow as tf
a = 10*tf.random_normal([4,4,4])
b = tf.reshape(a,[2,2,4,4])
c = tf.reshape(b,[4,4,4])
d = c-a
with tf.Session() as sess:
aa = sess.run(d)
print(aa)
輸出為一個全0矩陣,證明reshape相同形狀經過同為逆的兩次reshape后,是能還原的,