Imported from https://github.com/labsix/adversarial-logit-pairing-analysis