TopK

TopK kelas akhir publik

Menemukan nilai dan indeks elemen terbesar `k` untuk dimensi terakhir.

Jika masukannya adalah vektor (peringkat-1), temukan `k` entri terbesar dalam vektor dan keluarkan nilai dan indeksnya sebagai vektor. Jadi `nilai[j]` adalah entri terbesar ke-`j` dalam `input`, dan indeksnya adalah `indeks[j]`.

Untuk matriks (masing-masing masukan dengan peringkat lebih tinggi), hitung entri `k` teratas di setiap baris (masing-masing vektor sepanjang dimensi terakhir). Dengan demikian,

nilai.bentuk = indeks.bentuk = masukan.bentuk[:-1] + [k]

Jika dua elemen sama, elemen dengan indeks lebih rendah akan muncul terlebih dahulu.

Kelas Bersarang

kelas

Opsi TopK

Atribut opsional untuk TopK

Konstanta

Rangkaian

OP_NAME

Nama operasi ini dikenal dengan mesin inti TensorFlow

Metode Publik

statis <T memperluas TNomber > TopK <T>	buat ( Lingkup lingkup, masukan Operan <T>, Operan < TInt32 > k, Opsi... opsi) Metode pabrik untuk membuat kelas yang membungkus operasi TopK baru.
Keluaran < TInt32 >	indeks () Indeks `nilai` dalam dimensi terakhir `input`.
Opsi TopK.statis	diurutkan (diurutkan secara Boolean)
Keluaran <T>	nilai () Elemen terbesar `k` sepanjang setiap irisan dimensi terakhir.

Metode Warisan

Dari kelas org.tensorflow.op.RawOp

boolean terakhir	sama dengan (Objek objek)
int terakhir	Kode hash ()
Operasi	op () Kembalikan unit komputasi ini sebagai satu Operation .
String terakhir	keString ()

Dari kelas java.lang.Object

boolean	sama dengan (Objek arg0)
Kelas terakhir<?>	dapatkan Kelas ()
ke dalam	Kode hash ()
kekosongan terakhir	memberitahu ()
kekosongan terakhir	beri tahuSemua ()
Rangkaian	keString ()
kekosongan terakhir	tunggu (arg0 panjang, int arg1)
kekosongan terakhir	tunggu (argumen panjang0)
kekosongan terakhir	Tunggu ()

Dari antarmuka org.tensorflow.op.Op

Lingkungan Eksekusi abstrak	env () Kembalikan lingkungan eksekusi tempat operasi ini dibuat.
Operasi abstrak	op () Kembalikan unit komputasi ini sebagai satu Operation .

Konstanta

String akhir statis publik OP_NAME

Nama operasi ini dikenal dengan mesin inti TensorFlow

Nilai Konstan: "TopKV2"

Metode Publik

public static TopK <T> buat ( Lingkup lingkup, masukan Operan <T>, Operan < TInt32 > k, Opsi... opsi)

Metode pabrik untuk membuat kelas yang membungkus operasi TopK baru.

Parameter

cakupan	ruang lingkup saat ini
memasukkan	1-D atau lebih tinggi dengan dimensi terakhir setidaknya `k`.
k	0-D. Jumlah elemen teratas yang harus dicari sepanjang dimensi terakhir (sepanjang setiap baris matriks).
pilihan	membawa nilai atribut opsional

Kembali

• contoh baru TopK

Output publik < TInt32 > indeks ()

Indeks `nilai` dalam dimensi terakhir `input`.

TopK.Options statis publik diurutkan (diurutkan Boolean)

Parameter

diurutkan	Jika benar, elemen `k` yang dihasilkan akan diurutkan berdasarkan nilai dalam urutan menurun.

Nilai keluaran <T> publik ()

Elemen terbesar `k` sepanjang setiap irisan dimensi terakhir.