warp.argmax#

warp.argmax(a: Vector[Any, Scalar]) → uint32#

  • Kernel

Compute the index of the maximum element of vector a.