Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

ApproxTopK

kelas akhir publik ApproxTopK

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>
buat ( Lingkup lingkup, input Operand <T>, Long k, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi ApproxTopK baru.
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.