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

tf.compat.v1.arg_max

Devuelve el índice con el valor más grande entre las dimensiones de un tensor.

Tenga en cuenta que en caso de empate no se garantiza la identidad del valor de retorno.

Uso:

 import tensorflow as tf
a = [1, 10, 26.9, 2.8, 166.32, 62.3]
b = tf.math.argmax(input = a)
c = tf.keras.backend.eval(b)
# c = 4
# here a[4] = 166.32 which is the largest element of a across axis 0
 

input Un Tensor . Debe ser uno de los siguientes tipos: float32 , float64 , int32 , uint8 , int16 , int8 , complex64 , int64 , qint8 , quint8 , qint32 , bfloat16 , uint16 , complex128 , half , uint32 , uint64 , bool .
dimension Un Tensor . Debe ser uno de los siguientes tipos: int32 , int64 . int32 o Int64, deben estar en el intervalo [-rank(input), rank(input)) . Describe que la dimensión de la entrada Tensor para reducir través. Para los vectores, dimensión uso = 0.
output_type Un opcional tf.DType de: tf.int32, tf.int64 . Por defecto es tf.int64 .
name Un nombre para la operación (opcional).

A Tensor de tipo output_type .