def tdp
import bendtorch@1.3.0.0/main.bend as Bendtorchimport Base
3499def tdp(-P: Type, g: P, +p1: F32, +p2: F32, +l: F32, rp: PL<P>, rt: PTU) -> TrainPL<P>:
3500 match rp rt:
3501 case PL{p, xp} PTU{t, +n}:
3502 tdp2(P, p, g, p1, p2, l, xp, pt_store(t, Array.new(F32, depth(U32.to_nat(n)), 0.0), 0))