Imported from https://github.com/cvqluu/Angular-Penalty-Softmax-Losses-Pytorch