chainerx.argmin

chainerx.argmin(a, axis=None)

Returns the indices of the minimum along an axis.

Parameters

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

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

Returns

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

Return type

ndarray

See also

numpy.argmin()