Find index of largest value in array

Is there a convenient single function that takes an INDArray (1D if necessary) and returns the index of the largest item? I’m sure there must be something but I can’t find it.


I think you are looking for argmax

That looks about right - I’ll go and experiment. Many thanks for the quick reply.