アクティベーション関数を適用する前に、各チャネルで CNN 内の画像を正規化したいと考えています。これは pytorch の InstanceNorm2d が行うことであり、軸が (N、H、W、C) であるためLayerNormalization、パラメーターでテンソルフローを使用することでこの動作を模倣できると考えました。axis=-1これは私の入力画像の形状 (1, 128, 128, 1) です。
y3 = keras.layers.LayerNormalization(axis=3)(x)
この方法で正規化を適用すると、出力イメージは次のようになります。
私はこの振る舞いを理解していません。誰かが私が間違っていることを知っていますか? どんな助けでも大歓迎です！