TopK kelas akhir publik
Menemukan nilai dan indeks elemen terbesar `k` untuk dimensi terakhir.
Jika masukannya adalah vektor (peringkat-1), temukan `k` entri terbesar dalam vektor dan keluarkan nilai dan indeksnya sebagai vektor. Jadi `nilai[j]` adalah entri terbesar ke-`j` dalam `input`, dan indeksnya adalah `indeks[j]`.
Untuk matriks (masing-masing masukan dengan peringkat lebih tinggi), hitung entri `k` teratas di setiap baris (masing-masing vektor sepanjang dimensi terakhir). Dengan demikian,
nilai.bentuk = indeks.bentuk = masukan.bentuk[:-1] + [k]
Jika dua elemen sama, elemen dengan indeks lebih rendah akan muncul terlebih dahulu.
Kelas Bersarang
| kelas | Opsi TopK | Atribut opsional untuk TopK | |
Konstanta
| Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
| statis <T memperluas TNomber > TopK <T> | |
| Keluaran < TInt32 > | indeks () Indeks `nilai` dalam dimensi terakhir `input`. |
| Opsi TopK.statis | diurutkan (diurutkan secara Boolean) |
| Keluaran <T> | nilai () Elemen terbesar `k` sepanjang setiap irisan dimensi terakhir. |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Nilai Konstan: "TopKV2"
Metode Publik
public static TopK <T> buat ( Lingkup lingkup, masukan Operan <T>, Operan < TInt32 > k, Opsi... opsi)
Metode pabrik untuk membuat kelas yang membungkus operasi TopK baru.
Parameter
| cakupan | ruang lingkup saat ini |
|---|---|
| memasukkan | 1-D atau lebih tinggi dengan dimensi terakhir setidaknya `k`. |
| k | 0-D. Jumlah elemen teratas yang harus dicari sepanjang dimensi terakhir (sepanjang setiap baris matriks). |
| pilihan | membawa nilai atribut opsional |
Kembali
- contoh baru TopK
TopK.Options statis publik diurutkan (diurutkan Boolean)
Parameter
| diurutkan | Jika benar, elemen `k` yang dihasilkan akan diurutkan berdasarkan nilai dalam urutan menurun. |
|---|