def sr2
import bend-ml-tensor-fork@0.1.6.0/main.bend as BendMlTensorForkimport Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA
330def sr2(+mm1: Nat, +o: U32, +label: Nat, +nf: F32, q: TA.AF) -> TA.L2:
331 match q:
332 case TA.AF{c, +s}:
333 sr3(mm1, o, label, s, nf, TA.cea_p(Nat.is_le(label, mm1), o, label, s, c))