def argmax2

bend-ml-tensor-array@0.1.19.0 · main.bend:491

import bend-ml-tensor-array@0.1.19.0/main.bend as BendMlTensorArray

import Base

491def argmax2(xs: List<&2, F32>, g: Bool, +i: Nat, +bi: Nat, +bv: F32) -> Nat:
492  match xs g:
493    case Nil{} _:
494      bi
495    case Con{+h, +t} True{}:
496      argmax2(t, gt_head(t, h), 1n+i, i, h)
497    case Con{h, +t} False{}:
498      argmax2(t, gt_head(t, bv), 1n+i, bi, bv)

Uses (5)

Used by (1)

bend-ml-tensor-array@0.1.19.0/main:argmax