argmax#

rbc.stdlib.searching_functions.argmax(x)#

Returns the indices of the maximum values along a specified axis.

Array-API ‘argmax’ doc