将一个4D张量整形为2D，其尺寸为无

Question

我有一个关于tf.reshape

的问题

E.g。形状为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的第一个维度是无。

我如何处理此案例，有任何建议吗？

Answer 1

你可以这样做：

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