chainer.functions.argmin

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

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

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

Output variable.
Return type:

Variable