tensorly.argmax

argmax(tensor)

The argument of the max value in a tensor.

Parameters
tensortensor
Returns
scalar