def step_of

mlbend-generative@0.1.1.0 · main.bend:361

import Base

361def step_of(+abar: F32, +abar_prev: F32, +eta: F32) -> Step:
362  +sig = sigma_of(abar, abar_prev, eta)
363  +c0 = (1.0 / F32.sqrt(abar) : F32)
364  +c1 = F32.sqrt(F32.max(0.0, (1.0 - abar : F32)))
365  +c2 = F32.sqrt(F32.max(0.0, abar_prev))
366  +c3 = F32.sqrt(F32.max(0.0, (1.0 - (abar_prev + (sig * sig : F32) : F32) : F32)))
367  Step{c0, c1, c2, c3, sig}