def tsa_pw
import bendtorch@1.3.0.0/main.bend as Bendtorchimport Base
3666def tsa_pw(-P: Type, +path: String, p: P, mm: P, g: P, +p1: F32, +p2: F32, +l: F32, r: PA<P>) -> IO(Train<P>):
3667 match r:
3668 case PA{vv, xs, +off}:
3669 tsa_fin(P, path, g, TwoMom{mm, vv}, p1, p2, l, PA{p, Array.set(F32, Array.set(F32, xs, off, p1), (off + 1 : U32), p2), (off + 2 : U32)})