def ln_s2

bendtorch@1.3.0.0 · main.bend:1827

import Base
import bend-ml-tensor-array@0.1.18.0/main.bend as TA

1827def ln_s2(-d: Nat, a: MatStep<1n, d>, c: MatStep<1n, d>) -> Step<LN<d>>:
1828  match a c:
1829    case MatStep{w, gw, sw} MatStep{b, gb, sb}:
1830      Step{LN{w, b}, LN{gw, gb}, OptState.map(Two<TA.Mat<1n, d>, TA.Mat<1n, d>>, LN<d>, OptState.join(TA.Mat<1n, d>, TA.Mat<1n, d>, sw, sb), x => ln_unpair(d, x), x => ln_unpair(d, x))}