cross_entropy
log_softmax
nll_loss
softmax