- am-cross-entropy-loss logits targetprocedure
Numerically stable softmax cross-entropy, averaged over the batch. logits is a (batch, C) lazy tensor of raw scores and target a (batch, C) lazy tensor of one-hot labels. Returns a scalar lazy tensor.
Numerically stable softmax cross-entropy, averaged over the batch. logits is a (batch, C) lazy tensor of raw scores and target a (batch, C) lazy tensor of one-hot labels. Returns a scalar lazy tensor.