~/jendhub

def pair_step2

bendtorch@1.3.0.0 · main.bend:2080

import bendtorch@1.3.0.0/main.bend as Bendtorch
import Base

2080# one optimizer step on two layers' parameters, of any types
2081def pair_step2(~P1: Type, ~X1: Type, ~Y1: Type, ~S1: Type, ~P2: Type, ~X2: Type, ~Y2: Type, ~S2: Type, ~la: Layer<P1, X1, Y1, S1>, ~lb: Layer<P2, X2, Y2, S2>, +op: Opt, p: Two<P1, P2>, g: Two<P1, P2>, st: Two<OptState<P1>, OptState<P2>>) -> Step<Two<P1, P2>>:
2082  match p g st:
2083    case Two{p1, p2} Two{g1, g2} Two{s1, s2}:
2084      pair_s3(P1, P2, Layer.step(P1, X1, Y1, S1, la, op, p1, g1, s1), Layer.step(P2, X2, Y2, S2, lb, op, p2, g2, s2))

Uses (7)

Used by (1)