Imported from https://github.com/fteufel/PyTorch-GRU-D