tensorflow中是否有API来减少维度？例如，给定张量[[1,2], [3,4], [5,6]]我想减少其最后一个维度，使其变为[1, 2, 3, 4, 5, 6]。
有一个API tf.expand_dims()可以做相反的事情。有tf.reduce_dims()吗？
答案 0 :(得分：1)
您可以使用tf.squeeze删除1-D尺寸，也可以使用tf.reshape更改其他兼容形状的张量形状
在您的情况下，由于您没有1-D尺寸，您可以使用重塑来获得您想要的内容。
假设a是您的输入张量，您可以＆＃34;展平＆＃34;输入使用：
b = tf.reshape(a, (-1,))