Loss

Contract for loss functions. Implementations should validate shapes/dtypes and return either a scalar (when reduced) or per-element loss tensor.

Functions

abstract fun <T : DType, V> forward(preds: Tensor<T, V>, targets: Tensor<out DType, *>, ctx: ExecutionContext, reduction: Reduction = Reduction.MEAN): Tensor<T, V>