Tensorflow 张量流模型的简化形状

所以我创建了一个tensorflow模型，但是它使用了一个现有的模型，其输出的形状是[1512]或者类似的东西。 我想在最后添加几层，但它需要以与类数相同的输出形状结束，而类数要小得多

[1,5]例如

减少输出以使其匹配而不丢失太多数据的最佳方法是什么

---

我尝试了池化，但我认为我理解得不够。

您可以将输出连接到密集层，如：logits=tf.layers.denseoutputs，units=5以将512个输出单位减少到5个。

谢谢！工作得很好！