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）。如何实施重塑？ None维度必须是第一个维度吗？

Answer 1

您可以使用tf.shape将形状作为整数Tensor。

所以在你的例子中：

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