Find index of largest value in array

ND4J
1

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.

Thanks

2

I think you are looking for argmax

3

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