Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

tf.compat.v1.arg_max

Mengembalikan index dengan nilai terbesar di dimensi tensor a.

Perhatikan bahwa dalam kasus ikatan identitas nilai kembali tidak dijamin.

Pemakaian:

 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 Sebuah Tensor . Harus menjadi salah satu jenis berikut: float32 , float64 , int32 , uint8 , int16 , int8 , complex64 , int64 , qint8 , quint8 , qint32 , bfloat16 , uint16 , complex128 , half , uint32 , uint64 , bool .
dimension Sebuah Tensor . Harus menjadi salah satu jenis berikut: int32 , int64 . int32 atau int64, harus berada dalam kisaran [-rank(input), rank(input)) . Menggambarkan yang dimensi input Tensor untuk mengurangi seluruh. Untuk vektor, penggunaan dimensi = 0.
output_type Opsional tf.DType dari: tf.int32, tf.int64 . Default untuk tf.int64 .
name Sebuah nama untuk operasi (opsional).

Sebuah Tensor jenis output_type .