Questa pagina è stata tradotta dall'API Cloud Translation.
Switch to English

tf.compat.v1.arg_max

Restituisce l'indice con il valore più grande attraverso le dimensioni di un tensore.

Si noti che in caso di parità l'identità del valore di ritorno non è garantito.

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 A Tensor . Deve essere uno dei seguenti tipi: float32 , float64 , int32 , uint8 , int16 , int8 , complex64 , int64 , qint8 , quint8 , qint32 , bfloat16 , uint16 , complex128 , half , uint32 , uint64 , bool .
dimension A Tensor . Deve essere uno dei seguenti tipi: int32 , int64 . Int32 o Int64, devono essere nell'intervallo [-rank(input), rank(input)) . Descrive quali dimensione dell'ingresso Tensor per ridurre tutti. Per vettori, l'uso dimensione = 0.
output_type Un optional tf.DType da: tf.int32, tf.int64 . Il valore predefinito è tf.int64 .
name Un nome per l'operazione (opzionale).

Un Tensor di tipo output_type .