Mengembalikan nilai unik TopK dalam array dalam urutan yang diurutkan. Itu
waktu berjalan sebanding dengan produk K dan ukuran input. Mengurutkan seluruh larik lebih efisien untuk nilai K yang cukup besar. Algoritma median-of-median mungkin lebih cepat, namun sulit diterapkan secara efisien di XLA. Jika angka uniknya kurang dari K (bukan NAN), hasilnya diisi dengan tak terhingga negatif. NaN tidak pernah dikembalikan. Angka-angka di bawah normal dipindahkan ke nol. Jika suatu elemen muncul di beberapa indeks, indeks tertinggi akan dikembalikan. Jika elemen TopK tidak pernah muncul di input karena nilai padding, indeks akan diisi dengan nilai negatif. Jika nilai padding muncul di input dan padding diperlukan, indeks tertinggi dari nilai padding akan dikembalikan. Semantiknya tidak sama dengan kth_order_statistic.
Konstanta
Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
TopKUnique statis | |
Keluaran < TFloat32 > | topk () |
Keluaran < TInt32 > | indeks topk () |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Metode Publik
public static TopKUnique buat ( Lingkup lingkup, Operan < TFloat32 > masukan, Panjang k)
Metode pabrik untuk membuat kelas yang membungkus operasi TopKUnique baru.
Parameter
cakupan | ruang lingkup saat ini |
---|
Kembali
- contoh baru TopKUnique