Mengembalikan nilai min/max k dan indeks operan inputnya secara perkiraan.
Lihat https://arxiv.org/abs/2206.14286 untuk detail algoritme. Operasi ini hanya dioptimalkan pada TPU saat ini.
Kelas Bersarang
kelas | ApproxTopK.Options | Atribut opsional untuk ApproxTopK |
Metode Publik
ApproxTopK.Options statis | agregatToTopk (Agregat BooleanToTopk) |
statis <T memperluas Angka> ApproxTopK <T> | |
Keluaran <Bilangan Bulat> | indeks () Indeks `nilai` di sepanjang `pengurangan_dimensi` dari operan `input`. |
ApproxTopK.Options statis | isMaxK (Boolean isMaxK) |
ApproxTopK.Options statis | recallTarget (Float recallTarget) |
ApproxTopK.Options statis | reduksiDimensi (Dimensi reduksi panjang) |
ApproxTopK.Options statis | reductionInputSizeOverride (Pengurangan panjangInputSizeOverride) |
Keluaran <T> | nilai () Nilai k min/maks di sepanjang `reduksi_dimensi` dari operan `input`. |
Metode yang Diwarisi
Metode Publik
publik statis ApproxTopK.Options agregatToTopk ( Boolean agregatToTopk)
Parameter
agregatToTopk | Jika benar, agregat perkiraan hasil ke top-k. Jika salah, kembalikan hasil perkiraan. Jumlah hasil perkiraan adalah implementasi yang ditentukan dan lebih besar sama dengan `k` yang ditentukan. |
---|
public static ApproxTopK <T> create ( Scope scope, Operand <T> input, Long k, Options... options)
Metode pabrik untuk membuat kelas yang membungkus operasi ApproxTopK baru.
Parameter
cakupan | lingkup saat ini |
---|---|
memasukkan | Array untuk mencari. Harus setidaknya 1-D dari tipe mengambang |
k | Menentukan jumlah min/max-k. |
pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru dari ApproxTopK
Output publik <Integer> indeks ()
Indeks `nilai` di sepanjang `pengurangan_dimensi` dari operan `input`.
ApproxTopK.Options publik statis adalahMaxK (Boolean isMaxK)
Parameter
isMaxK | Jika benar, hitung max-k; jika tidak, hitung min-k. |
---|
public static ApproxTopK.Options recallTarget (Float recallTarget)
Parameter
ingatTarget | Ingat target untuk aproksimasi. Rentang dalam (0,1] |
---|
public static ApproxTopK.Options reductionDimension (Long reductionDimension)
Parameter
penguranganDimensi | Dimensi bilangan bulat yang akan ditelusuri. Standar: -1. |
---|
public static ApproxTopK.Options reductionInputSizeOverride (Pengurangan panjangInputSizeOverride)
Parameter
penguranganInputSizeOverride | Jika disetel ke nilai positif, ukuran yang ditentukan oleh `input[reduction_dim]` akan ditimpa oleh ukuran yang ditentukan untuk mengevaluasi penarikan. Opsi ini berguna ketika `input` yang diberikan hanya merupakan subset dari keseluruhan komputasi dalam SPMD atau pipeline terdistribusi, di mana ukuran input sebenarnya tidak dapat ditangguhkan oleh bentuk `input`. |
---|
Output publik <T> nilai ()
Nilai k min/maks di sepanjang `reduksi_dimensi` dari operan `input`. Dimensinya sama dengan operan `input` kecuali untuk `reduction_dimension`: ketika `aggregate_to_topk` bernilai true, dimensi reduksinya adalah `k`; jika tidak, lebih besar sama dengan `k` di mana ukurannya ditentukan oleh implementasi.