例如：
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)