TorchLean API

Docs HomeGuideExamplesGraphs

NN.Proofs.Autograd.Tape.Nodes.Losses.MSE

Mean-Squared Error #

Tape node and Fréchet derivative proof for scalar mean-squared error.

source
noncomputable def Proofs.Autograd.TapeNodes.mseLoss {Γ : List Spec.Shape} {s : Spec.Shape} (yhat target : Idx Γ s) :
Node Γ Spec.Shape.scalar

Mean-squared-error loss node: c * ‖yhat - target‖^2, with c = 1 / size(s).

Instances For
    source
    noncomputable def Proofs.Autograd.TapeNodes.mseLossFderiv {Γ : List Spec.Shape} {s : Spec.Shape} (yhat target : Idx Γ s) :
    NodeFDerivCorrect (mseLoss yhat target)

    NodeFDerivCorrect for mse_loss.

    Instances For