Numpy's argmax() in Swift

Question

I\'m working on a project which converts user\'s face to emoji. I use Apple\'s ARKit in this purpose.

I need to get the most probable option. I wrote this code:

<