Imported from https://github.com/cheonbok94/Pytorch-Relational-Recurrent-Neural-networks