def tr_adam2

bendtorch@1.3.0.0 · main.bend:3538

import bendtorch@1.3.0.0/main.bend as Bendtorch
import Base

3538def tr_adam2(-P: Type, p: P, mm: P, vv: P, g: P, p1: F32, r: Array<F32> & F32) -> Train<P>:
3539  match r:
3540    case Tuple{xs, p2}: Train{p, g, TwoMom{mm, vv}, p1, p2, 0.0}

Uses (6)

Used by (1)