Imported from https://github.com/mtkwT/Hessian-based-analysis-tensorflow