Imported from https://github.com/amirhfarzaneh/lsoftmax-pytorch