mseLoss method

Tensor mseLoss(
  1. Tensor target
)

Implementation

Tensor mseLoss(Tensor target) {
  final diff = this - target;
  final squared = diff.pow(2.0);

  // Use sum instead of mean if your mean is broken
  final totalSum = squared.sum();

  // Manually normalize the loss value
  final scalarLoss = totalSum / (length).toDouble();

  // tracker.addAll([diff, squared, totalSum]);
  return scalarLoss;
}