Imported from https://github.com/hayoung-kim/tf2-deep-ensemble-uncertainty