减少张量流模型的形状

Question

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

[1，5]。

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

我尝试了池，但我不认为我足够理解它。

Answer 1

您可以将输出连接到如下的致密层：logits = tf.layers.dense(outputs, units=5)以将输出单元从512减少到5。