LossFunction

All Known Implementing Classes:: CrossEntropyLoss, MSELoss

public interface LossFunction

Method Summary

Modifier and Type

Method

Description

Tensor

gradient(Tensor predicted, Tensor actual)

double

loss(Tensor predicted, Tensor actual)

Method Details
- loss
  
  double loss(Tensor predicted, Tensor actual)
- gradient
  
  Tensor gradient(Tensor predicted, Tensor actual)