chainerx.argmax

chainerx.argmax(a, axis=None)

Returns the indices of the maximum along an axis.

Parameters

  • a (ndarray) – Array to take the indices of the maximum of.

  • axis (None or int) – Along which axis to compute the maximum. The flattened array is used by default.

Returns

The indices of the maximum of a, along the axis if specified.

Return type

ndarray

See also

numpy.argmax()