Imported from https://github.com/morris-frank/nsynth-pytorch