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