def mse_fin

bendtorch@1.3.0.0 · main.bend:3264

import bendtorch@1.3.0.0/main.bend as Bendtorch

import Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA

3264def mse_fin(-n: Nat, -c: Nat, +m: U32, +nc: F32, r: D3S) -> LossOut<TA.Mat<n, c>>:
3265  match r:
3266    case D3S{y, t, q}:
3267      LossOut{(tree_sum(m, q) / nc : F32), TA.Mat{y}}

Uses (8)

Used by (1)

bendtorch@1.3.0.0/main:MSE