LossFunctions
object LossFunctions
Type members
Classlikes
case class BCEWithLogits(posWeights: Option[STen], reduction: Reduction, ignore: Long) extends LossFunction
case class NLL(numClasses: Int, classWeights: STen, reduction: Reduction, ignore: Long) extends LossFunction
Return a loss function which takes outputs of time step x batch x classes and targets of time step x batch The returned loss is averaged over the batch and the time steps
Return a loss function which takes outputs of time step x batch x classes and targets of time step x batch The returned loss is averaged over the batch and the time steps