Imported from https://github.com/fshamshirdar/pytorch-rdpg