tf.reshape

在做多張的圖片為單個輸入的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后,是能還原的,

?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內容

友情鏈接更多精彩內容