Fork me on GitHub

src/arraymancer/nn/loss/mean_square_error_loss

Search:
Group by:
  Source Edit

Types

MSELoss[TT] {.final.} = ref object of Gate[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

Made with Nim. Generated: 2024-05-12 14:01:08 UTC
Arraymancer Technical reference Tutorial Spellbook (How-To's) Under the hood