Ta strona została przetłumaczona przez Cloud Translation API.
Switch to English

tf.compat.v1.arg_max

Zwraca indeks o największej wartości całej wymiarach tensora.

Należy pamiętać, że w przypadku więzi tożsamość wartości zwracanej nie jest gwarantowana.

Stosowanie:

 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 Tensor . Musi to być jeden z następujących typów: float32 , float64 , int32 , uint8 , int16 , int8 , complex64 , int64 , qint8 , quint8 , qint32 , bfloat16 , uint16 , complex128 , half , uint32 , uint64 , bool .
dimension Tensor . Musi to być jeden z następujących typów: int32 , int64 . Int32 lub Int64, musi mieścić się w zakresie [-rank(input), rank(input)) . Opisuje których wymiar wejściowego tensora celu zmniejszenia średnicy. Wektory wymiar zastosowanie = 0.
output_type Opcjonalny tf.DType od: tf.int32, tf.int64 . Domyślnie tf.int64 .
name Nazwę operacji (opcjonalnie).

Tensor typu output_type .