//clika-runtime/io.clika.runtime/Ops/crossEntropy

crossEntropy

[common]
fun crossEntropy(input: Tensor, target: Tensor, weight: Tensor? = null, ignoreIndex: Long? = null, reduction: Reduction = Reduction.MEAN): Tensor

crossEntropy(input: Tensor, target: Tensor, weight: Tensor? = null, ignoreIndex: Long? = null, reduction: Reduction = Reduction.MEAN): the cross_entropy operator. Cross-entropy loss over class LOGITS; the class dim is LAST. Equivalent to log_softmax over the class axis followed by negative-log-likelihood: