8

tf.keras を使用して再帰型ニューラル ネットワークに層の正規化を適用したいと考えています。TensorFlow 2.0 では、 にLayerNormalizationクラスがありますが、各時間ステップで (使用されるように設計されているため)のような再帰レイヤーtf.layers.experimentalで使用する方法が不明です。カスタム セルを作成する必要がありますか、それとももっと簡単な方法がありますか?LSTM

たとえば、各タイム ステップでドロップアウトを適用するのは、レイヤーrecurrent_dropoutを作成するときに引数を設定するのと同じくらい簡単ですが、引数LSTMはありませんrecurrent_layer_normalization

4

2 に答える 2

4

tensorflow アドオンには、すぐに使えるビルド済みのLayerNormLSTMCellがあります。

詳細については、このドキュメントを参照してください。tensorflow-addonsこのセルをインポートする前に、インストールする必要がある場合があります。

pip install tensorflow-addons
于 2020-06-24T03:16:04.803 に答える