Mengembalikan indeks dengan nilai terbesar di seluruh dimensi tensor.
Perhatikan bahwa jika terjadi ikatan, identitas nilai yang dikembalikan tidak dijamin.
Penggunaan:
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
Konstanta
Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
Keluaran <V> | sebagai Keluaran () Mengembalikan pegangan simbolis tensor. |
statis <V memperluas TNumber > ArgMax <V> | |
ArgMax statis <TInt64> | |
Keluaran <V> | keluaran () |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Metode Publik
Keluaran publik <V> sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
public static ArgMax <V> buat ( Lingkup lingkup , Operan <? extends TType > input, Operand <? extends TNumber > dimensi, Kelas<V> outputType)
Metode pabrik untuk membuat kelas yang membungkus operasi ArgMax baru.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
dimensi | int32 atau int64, harus berada dalam rentang `[-rank(input), rank(input))`. Menjelaskan dimensi Tensor masukan mana yang akan dikurangi. Untuk vektor, gunakan dimensi = 0. |
Kembali
- contoh baru ArgMax
public static ArgMax < TInt64 > buat ( Lingkup lingkup , Operan <? extends TType > input, Operand <? extends TNumber > dimensi)
Metode pabrik untuk membuat kelas yang membungkus operasi ArgMax baru menggunakan tipe keluaran default.
Parameter
cakupan | ruang lingkup saat ini |
---|---|
dimensi | int32 atau int64, harus berada dalam rentang `[-rank(input), rank(input))`. Menjelaskan dimensi Tensor masukan mana yang akan dikurangi. Untuk vektor, gunakan dimensi = 0. |
Kembali
- contoh baru ArgMax