def tr1
import bendtorch@1.3.0.0/main.bend as Bendtorchimport Base
3555def tr1(~P: Type, ~X: Type, ~Y: Type, ~S: Type, ~m: Layer<P, X, Y, S>, mm: P, vv: P, g: P, r: PA<P>) -> Train<P>:
3556 match r:
3557 case PA{p, xs, off}:
3558 tr2(~P, ~X, ~Y, ~S, ~m, p, vv, g, Layer.load(P, X, Y, S, m, mm, xs, off))