本文介绍如何在TensorFlow中使一个张量的形状与另一个张量相同，通过使用tf.reshape函数并结合目标张量的shape属性实现。具体示例为将tensorA的形状调整为与tensorB一致。
例如：
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 的形状
最新推荐文章于 2025-02-02 21:37:03 发布
例如：
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) 
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
127
