def mse_squared

bendtorch-core@0.1.0.0 · src/autograd/loss.bend:214

import Base
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core

214def mse_squared(e: Core.Dynamic, pair: (Core.Dynamic & Core.Dynamic) & Result<&2, &1, Core.Error, Core.Dynamic>) ->
215  Result<&2, &1, Error, Recorded>:
216  (operands, result) = pair
217  (kept, copy) = operands
218  match result:
219    case Fail{error}:
220      Fail{TensorError{error}}
221    case Done{squares}:
222      recorded(e, kept, mean(squares))