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

tf.math.argmax

Tensorflow 1 versione Visualizza sorgente su GitHub

Restituisce l'indice con il valore più grande tra gli assi di un tensore.

Utilizzato nei notebook

Utilizzato nella guida Utilizzato nelle esercitazioni

In caso di identità restituisce l'indice più piccolo.

Per esempio:

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 A Tensor .
axis Un numero intero, l'asse di ridurre tutti. Predefinito a 0.
output_type Un DTYPE uscita opzionale ( tf.int32 o tf.int64 ). Il valore predefinito è tf.int64 .
name Un nome opzionale per l'operazione.

Un Tensor di tipo output_type .