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> | |
| 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.