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
or N-dimensional array) – 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