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