argmax¶

dragon.vm.tensorflow.math.argmax(
  input,
  axis=None,
  name=None,
  **kwargs
)[source]¶

Compute the index of maximum elements along the given axis.

The argument axis could be negative or None:

x = tf.constant([[1, 2, 3], [4, 5, 6]])

# A negative ``axis`` is the last-k axis
print(tf.math.argmax(x, 1))
print(tf.math.argmax(x, -1))  # Equivalent

# If ``axis`` is None, the vector-style reduction
# will be applied to return a scalar index
print(tf.math.argmax(x))

Parameters:

• input (dragon.Tensor) – The input tensor.
• axis (int, optional) – The axis to reduce.
• name (str, optional) – The operation name.

Returns:

dragon.Tensor – The index of maximum elements.