Se usó la API de Cloud Translation para traducir esta página.
Switch to English

tf.math.argmax

TensorFlow 1 versión Ver código fuente en GitHub

Devuelve el índice con el valor más grande a través de ejes de un tensor.

Se utiliza en los cuadernos

Se utiliza en la guía Se utiliza en los tutoriales

En caso de identidad devuelve el índice más pequeño.

Por ejemplo:

A = tf.constant([2, 20, 30, 3, 6])
tf.math.argmax(A)  # A[2] is maximum in tensor A
<tf.Tensor: shape=(), dtype=int64, numpy=2>
B = tf.constant([[2, 20, 30, 3, 6], [3, 11, 16, 1, 8],
                 [14, 45, 23, 5, 27]])
tf.math.argmax(B, 0)
<tf.Tensor: shape=(5,), dtype=int64, numpy=array([2, 2, 0, 2, 2])>
tf.math.argmax(B, 1)
<tf.Tensor: shape=(3,), dtype=int64, numpy=array([2, 2, 1])>
C = tf.constant([0, 0, 0, 0])
tf.math.argmax(C) # Returns smallest index in case of ties
<tf.Tensor: shape=(), dtype=int64, numpy=0>

input Un Tensor .
axis Un número entero, el eje de reducir a través. Por defecto a 0.
output_type Un dtype de salida opcional ( tf.int32 o tf.int64 ). Por defecto es tf.int64 .
name Un nombre opcional para la operación.

A Tensor de tipo output_type .