テンソルフローモデルの形状を小さくする

Question

テンソルフローモデルを作成していますが、その出力が[1、512]のような形をしている既存のモデルを使用しています。 最後にいくつかのレイヤーを追加したいが、もっと小さなもののクラス数と同じ出力形状にする必要がある。

[1、5]

あまりにも多くのデータを失うことなく出力が一致するように出力を減らす最良の方法は何ですか？

私はプールしようとしましたが、十分理解していないと思います。

Answer 1

出力をlogits = tf.layers.dense(outputs, units=5)のような密なレイヤーに接続して、512から5出力単位に減らすことができます。