# 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