【发布时间】：2018-05-25 12:05:54
【问题描述】：
tensorflow 中是否有任何 API 可以降低维度？例如，给定一个张量[[1,2], [3,4], [5,6]]，我想减少它的最后一个维度，使其变为[1, 2, 3, 4, 5, 6]。
有一个 API tf.expand_dims() 可以做相反的事情。有tf.reduce_dims()吗？
【问题讨论】：
标签： tensorflow
【发布时间】：2018-05-25 12:05:54
【问题描述】：
您可以使用tf.squeeze 删除1-D 维度，或者您可以使用tf.reshape 将张量的形状更改为另一个兼容的形状
在你的情况下，由于你没有1-D 尺寸，你可以使用 reshape 来得到你想要的。
假设a 是您的输入张量，您可以使用以下方法“展平”输入：
b = tf.reshape(a, (-1,))
【讨论】：
-
我的经验是
tf.squeeze()通过返回形状不确定的张量而引起麻烦。问题在于，对于 CNN，人们通常不知道下一层的确切尺寸。我在这里找到了更好的答案：stackoverflow.com/questions/43336274/…