aliran tensor:: operasi:: TopK

Menemukan nilai dan indeks k elemen terbesar untuk dimensi terakhir.

Ringkasan

Jika masukannya adalah vektor (peringkat-1), carilah k entri terbesar dalam vektor tersebut dan keluarkan nilai dan indeksnya sebagai vektor. Jadi values[j] adalah entri terbesar ke- j di input , dan indeksnya adalah indices[j] .

Untuk matriks (masing-masing masukan dengan peringkat lebih tinggi), hitung k entri teratas di setiap baris (masing-masing vektor sepanjang dimensi terakhir). Dengan demikian,

values.shape = indices.shape = input.shape[:-1] + [k]

Jika dua elemen sama, elemen dengan indeks lebih rendah akan muncul terlebih dahulu.

Argumen:

• ruang lingkup: Objek Lingkup
• input: 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).

Atribut opsional (lihat Attrs ):

• diurutkan: Jika benar, k elemen yang dihasilkan akan diurutkan berdasarkan nilai dalam urutan menurun.

Pengembalian:

• Nilai Output : k elemen terbesar sepanjang setiap potongan dimensi terakhir.
• Indeks Output : Indeks values dalam dimensi terakhir input .

Atribut publik

::tensorflow::Output indices

Operation operation

::tensorflow::Output values

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k
)

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k,
  const TopK::Attrs & attrs
)

Attrs Sorted(
  bool x
)