Imported from https://github.com/atilberk/evidential-deep-learning-to-quantify-classification-uncertainty