nn.loss #

fn loss_loss[T](loss types.Loss, input &autograd.Variable[T], target &vtl.Tensor[T]) !&autograd.Variable[T]

fn mse_loss[T]() &MSELoss[T]

fn sigmoid_cross_entropy_loss[T]() &SigmoidCrossEntropyLoss[T]

fn softmax_cross_entropy_loss[T]() &SoftmaxCrossEntropyLoss[T]

fn (_ &MSELoss[T]) loss(input &autograd.Variable[T], target &vtl.Tensor[T]) !&autograd.Variable[T]

fn (_ &SigmoidCrossEntropyLoss[T]) loss(input &autograd.Variable[T], target &vtl.Tensor[T]) !&autograd.Variable[T]

fn (_ &SoftmaxCrossEntropyLoss[T]) loss(input &autograd.Variable[T], target &vtl.Tensor[T]) !&autograd.Variable[T]

struct MSELoss[T] {}

MSELoss

struct SigmoidCrossEntropyLoss[T] {}

SigmoidCrossEntropyLoss

struct SoftmaxCrossEntropyLoss[T] {}

SoftmaxCrossEntropyLoss