Imported from https://github.com/github-jnauta/pytorch-pne