argmin¶

dragon.math.argmin(
  inputs,
  axis=0,
  keepdims=False,
  **kwargs
)[source]¶

Compute the index of minimum elements along the given axis.

axis could be negative:

# A negative axis is the last-k axis
x = dragon.constant([[1, 2, 3], [4, 5, 6]])
print(dragon.math.argmin(x, axis=1))
print(dragon.math.argmin(x, axis=-1))  # Equivalent

Parameters:

• inputs (dragon.Tensor) – The input tensor.
• axis (int, optional, default=0) – The axis to reduce.
• keepdims (bool, optional, default=False) – Keep the reduced dimension or not.

Returns:

dragon.Tensor – The index of minimum elements.