減少張量流模型的形狀

Question

所以我創建了一個張量流模型，但它使用了一個現有的模型，其輸出形狀爲[1,512]或類似的東西。 我想在最後添加幾層，但它需要以與更小的類的數量相同的輸出形狀結束，例如

[1，5]。

什麼是減少輸出的最佳方式，使其匹配而不會丟失太多數據？

我嘗試了池，但我不認爲我足夠理解它。

Answer 1

您可以將輸出連接到如下的緻密層：logits = tf.layers.dense(outputs, units=5)以將輸出單元從512減少到5。