Tensorflow 将一个4D张量重塑为维度为零的2D张量

我有一个关于tf.重塑的问题

例如，一个张量

t1

，其形状为

[None，h，c，w]

。我想把张量重塑成2D，就像：

shape = t1.get_shape().as_list()
t2 = tf.reshape(t1, [shape[0]*shape[1], shape[2]*shape[3]])

但是，

t1

的第一个维度是无

我如何处理这个案件，有什么建议吗？

您可以做：

shape = tf.shape(t1)
t2 = tf.reshape(t1, [-1, shape[2]*shape[3]])