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])