将一个 4D 张量重塑为维度为 None 的 2D答案

【问题标题】：Reshape one 4D-tensor into 2D whose dimension is None将一个 4D 张量重塑为维度为 None 的 2D
【发布时间】：2018-10-26 09:11:19
【问题描述】：

我有一个关于tf.reshape的问题

例如一个张量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 的第一个维度是无。

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

【问题讨论】：

【解决方案1】：

你可以这样做：

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

【讨论】：