def Vec4.argmax.pick
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
103# the index (0-3) and value of the largest element; the first wins ties
104def Vec4.argmax.pick(c: Bool, +i: U32, +x: F32, +j: U32, +y: F32) -> U32 & F32:
105 match c:
106 case True{}:
107 (i, x)
108 case False{}:
109 (j, y)