Imported from https://github.com/dennis-sell/pytorch-fun