Imported from https://github.com/blacKitten13/Dropout-for-Model-Uncertainty