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

Dari kelas org.tensorflow.op.PrimitiveOp

boolean akhir	sama (objek objek)
akhir	kode hash ()
Operasi	op () Mengembalikan Operation yang mendasarinya
string terakhir	toString ()

Dari kelas java.lang.Object

boolean	sama dengan (Objek arg0)
Kelas terakhir<?>	dapatkanKelas ()
ke dalam	kode hash ()
kekosongan akhir	beri tahu ()
kekosongan akhir	beri tahu Semua ()
Rangkaian	toString ()
kekosongan akhir	tunggu (arg panjang0, int arg1)
kekosongan akhir	tunggu (arg panjang0)
kekosongan akhir	tunggu ()

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.