Imported from https://github.com/google/n-digit-mnist