chainer.functions.argmax

chainer.functions.argmax(x, axis=None)[source]

Returns index which holds maximum of array elements over a given axis.

Parameters:
  • x (Variable) – Array to find maximum elements.
  • axis (None or int) – Axis over which a max is performed. The default (axis = None) is perform a max over all the dimensions of the input array.
Returns:

Output variable.

Return type:

Variable