Fork me on GitHub
Arraymancer Technical reference Tutorial Spellbook (How-To's) Under the hood

mean_square_error_loss

Search:
Group by:

Types

MSELoss {...}{.final.}[TT] = ref object of Gate[TT]
  target: TT
  cache: Variable[TT]
  Source Edit

Procs

proc mse_loss[TT](input: Variable[TT]; target: TT): Variable[TT]
Mean square error loss function.
Input:
  • An input variable of predicted values of shape [batch_size, features]
  • The ground truth of the same shape
  Source Edit