def tr3

bendtorch@1.3.0.0 · main.bend:3546

import bendtorch@1.3.0.0/main.bend as Bendtorch

import Base

3546def tr3(-P: Type, p: P, mm: P, g: P, r: PA<P>) -> Train<P>:
3547  match r:
3548    case PA{vv, xs, +off}: tr_adam(P, p, mm, vv, g, Array.get(F32, xs, off), (off + 1 : U32))

Uses (7)

Used by (1)

bendtorch@1.3.0.0/main:tr2