处理批处理时重塑张量

时间:2019-05-28 13:10:55

标签: python tensorflow

我想调整称为X_rec的张量的大小。当前大小为(None, 151, 18921),我希望它为(None, 18921, 151)

张量具有第一个None大小,因为之后它会与要训练的批次一起调用。

我曾尝试使用tf.reshape,但收到错误消息“没有一个不是对形状有效的整数”

我正在从堆叠(列方式)列表中获取张量为(None, 18921)的151个列张量的张量:

X_rec = tf.stack(reconstructed,axis=1,name='X_rec')

我也尝试过做:

X_rec = tf.transpose(tf.stack(reconstructed,axis=1,name='X_rec'))

但是我也得到一个错误...

有什么想法吗?预先感谢!

0 个答案:

没有答案