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)