BendTensors.VecT.argmax: def of bend-tensors@0.0.0.2
Import: import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensors
Lines 610-619 of bend-tensors@0.0.0.2/bend_tensors.bend: https://jend.leeeet.dev/read.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&start_line=610&end_line=619

import Base

# the index and value of the largest element; the first wins ties
def VecT.argmax(+n: Nat, v: VecT(n)) -> U32 & F32:
  match n:
    case 0n:
      VecL{x} = v
      Vec4.argmax(x)
    case 1n+p:
      VecN{l, r} = v
      a b = VecT.argmax(p, l) VecT.argmax(p, r)
      VecT.argmax.two(U32.from_nat(VecT.size(p)), a, b)

Uses (11):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Pair
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32.from_nat
  https://jend.leeeet.dev/definition.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&name=Vec4.argmax
  https://jend.leeeet.dev/definition.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&name=VecT
  https://jend.leeeet.dev/definition.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&name=VecT.Leaf
  https://jend.leeeet.dev/definition.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&name=VecT.Node
  https://jend.leeeet.dev/definition.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&name=VecT.argmax.two
  https://jend.leeeet.dev/definition.txt?path=bend-tensors@0.0.0.2/bend_tensors.bend&name=VecT.size