Tensorflow可变长度

Question

重塑我用下面的重塑在我的模型：

data = tf.placeholder(tf.float32, shape = (BATCH_SIZE, N_CHUNK, WIN * N_SENSOR)) 

data_flattened = tf.reshape(data, [BATCH_SIZE*N_CHUNK, WIN*N_SENSOR]) 

现在我想N_CHUNK是可变的（即在尺寸NONE）。我如何实施重塑？无维必须是第一维吗？

Answer 1

您可以使用tf.shape来获得整形张量的形状。

所以在你的例子：

data = tf.placeholder(tf.float32, shape = (BATCH_SIZE, None, WIN * N_SENSOR)) 

data_flattened = tf.reshape(data, [BATCH_SIZE*tf.shape(data)[1], WIN*N_SENSOR])