ApproxTopK

kelas akhir publik ApproxTopK

Mengembalikan nilai min/maks k dan indeks operan masukannya dengan cara perkiraan.

Lihat https://arxiv.org/abs/2206.14286 untuk detail algoritme. Operasi ini hanya dioptimalkan pada TPU saat ini.

Kelas Bersarang

kelas

Kira-kiraTopK.Options

Atribut opsional untuk ApproxTopK

Metode Publik

ApproxTopK.Options statis	agregatToTopk (Agregat BooleanToTopk)
statis <T memperluas Nomor> Kira-kiraTopK <T>	buat ( Lingkup lingkup, masukan Operan <T>, Panjang k, Opsi... opsi) Metode pabrik untuk membuat kelas yang membungkus operasi ApproxTopK baru.
Keluaran <Bilangan Bulat>	indeks () Indeks `nilai` di sepanjang `dimensi_reduksi` dari operan `input`.
ApproxTopK.Options statis	isMaxK (Boolean isMaxK)
ApproxTopK.Options statis	recallTarget (Float recallTarget)
ApproxTopK.Options statis	penguranganDimensi (Dimensi pengurangan panjang)
ApproxTopK.Options statis	penguranganInputSizeOverride (Pengurangan panjangInputSizeOverride)
Keluaran <T>	nilai () Nilai min/maks k sepanjang `reduction_dimension` dari operan `input`.

Metode Warisan

Dari kelas org.tensorflow.op.PrimitiveOp

boolean terakhir	sama dengan (Objek objek)
int terakhir	Kode hash ()
Operasi	op () Mengembalikan Operation yang mendasarinya
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 ()

Metode Publik

public static ApproxTopK.Options agregatToTopk (Agregat BooleanToTopk)

Parameter

agregatToTopk	Jika benar, agregat memperkirakan hasil ke top-k. Jika salah, mengembalikan hasil perkiraan. Jumlah hasil perkiraan ditentukan oleh implementasi dan lebih besar dari `k` yang ditentukan.

public static ApproxTopK <T> buat ( Lingkup lingkup, input Operan <T>, Long k, Opsi... opsi)

Metode pabrik untuk membuat kelas yang membungkus operasi ApproxTopK baru.

Parameter

cakupan	ruang lingkup saat ini
memasukkan	Larik untuk mencari. Setidaknya harus 1-D dari tipe mengambang
k	Menentukan jumlah min/maks-k.
pilihan	membawa nilai atribut opsional

Kembali

• contoh baru ApproxTopK

Indeks Output publik <Integer> ()

Indeks `nilai` di sepanjang `dimensi_reduksi` dari operan `input`.

ApproxTopK.Options isMaxK statis publik (Boolean isMaxK)

Parameter

adalahMaxK	Jika benar, hitung max-k; jika tidak, hitung min-k.

ApproxTopK.Options recallTarget statis publik (Float recallTarget)

Parameter

ingatTarget	Ingat target untuk perkiraan. Rentang dalam (0,1]

Perkiraan statis publikTopK.Dimensi pengurangan opsi (Dimensi pengurangan panjang)

Parameter

penguranganDimensi	Dimensi integer yang akan digunakan untuk mencari. Bawaan: -1.

public static ApproxTopK.Options penguranganInputSizeOverride (Pengurangan panjangInputSizeOverride)

Parameter

penguranganInputSizeOverride	Jika disetel ke nilai positif, nilai tersebut akan menggantikan ukuran yang ditentukan oleh `input[reduction_dim]` untuk mengevaluasi penarikan kembali. Opsi ini berguna ketika `input` yang diberikan hanya sebagian dari keseluruhan komputasi di SPMD atau pipeline terdistribusi, yang mana ukuran input sebenarnya tidak dapat ditangguhkan oleh bentuk `input`.

Nilai keluaran <T> publik ()

Nilai min/maks k sepanjang `reduction_dimension` dari operan `input`. Dimensinya sama dengan operan `input` kecuali `dimensi_reduksi`: jika `agregat_ke_topk` benar, dimensi reduksinya adalah `k`; jika tidak, nilainya lebih besar sama dengan `k` yang ukurannya ditentukan oleh implementasi.