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

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.