ArgMax

ArgMax kelas akhir publik

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>
buat ( Lingkup lingkup , Operan <? extends TType > input, Operand <? extends TNumber > dimensi, Kelas<V> outputType)
Metode pabrik untuk membuat kelas yang membungkus operasi ArgMax baru.
ArgMax statis <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.
Keluaran <V>

Metode Warisan

Konstanta

String akhir statis publik OP_NAME

Nama operasi ini dikenal dengan mesin inti TensorFlow

Nilai Konstan: "ArgMax"

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

Keluaran publik <V> keluaran ()