Imported from https://github.com/T-Almeida/tensorflow-keras-multidimensional-rnn