例如：
tensor b的形狀是(?, 128, 160, 6)，想讓tensor A的形狀和b的一樣。
dim=[-1]+b.shape.as_list()[1:] #b的形狀變成（-1，128，160，6）
a=tf.reshape(a,shape=dim) tensorflow：如何將tensor A 的形狀reshape爲tensor b 的形狀
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
例如：
tensor b的形狀是(?, 128, 160, 6)，想讓tensor A的形狀和b的一樣。
dim=[-1]+b.shape.as_list()[1:] #b的形狀變成（-1，128，160，6）
a=tf.reshape(a,shape=dim)