减少张量流模型的形状

Question

所以我正在创建一个张量流模型，但它使用的是现有模型，其输出的形状为[1,512]或类似的东西。 我想在最后添加几层，但它需要最终的输出形状与类的数量相比更小

例如，

[1,5]。

减少输出的最佳方法是什么，以便匹配而不会丢失太多数据？

我尝试合并，但我认为我不太了解它。

Answer 1

您可以将输出连接到密集层，如：logits = tf.layers.dense(outputs, units=5)，以便从512减少到5个输出单位。