def mse_l
bendtorch@1.3.0.0 · main.bend:3256
import bendtorch@1.3.0.0/main.bend as Bendtorch
import Base
3256
3257def mse_l(k: Nat, +idx: U32, +kk: F32, q: Array<F32>, ry: Array<F32> & F32, rt: Array<F32> & F32) -> D3S:
3258 match k ry rt:
3259 case 0n Tuple{y, +a} Tuple{t, +b}:
3260 D3S{Array.set(F32, y, idx, ((a - b : F32) * kk : F32)), t, Array.set(F32, q, idx, ((a - b : F32) * (a - b : F32) : F32))}
3261 case 1n+n Tuple{y, +a} Tuple{t, +b}:
3262 mse_l(n, (idx + 1 : U32), kk, Array.set(F32, q, idx, ((a - b : F32) * (a - b : F32) : F32)), Array.get(F32, Array.set(F32, y, idx, ((a - b : F32) * kk : F32)), (idx + 1 : U32)), Array.get(F32, t, (idx + 1 : U32)))