custom_tf_layers

Author - Oliver Waring

Custom Model Layers for Keras & Tensorflow 2.0

LayerNormLSTMCell

LSTM Cell with layer normalization applied to features before gate activations

LayerNormRNNCell [1]

RNN Cell with layer normalization applied to features before output activation

[1] based on example from Hands-On Machine Learning by Aurélien Géron