mbollmann

# Source:
# https://github.com/farizrahman4u/seq2seq/blob/master/seq2seq/layers/state_transfer_lstm.py

from keras import backend as K
from keras.layers.recurrent import LSTM

class StateTransferLSTM(LSTM):
    """LSTM with the ability to transfer its hidden state.

    This layer behaves just like an LSTM, except that it can transfer (or