tensorflow::ops::TopK

Menemukan nilai dan indeks dari k elemen terbesar untuk dimensi terakhir.

Ringkasan

Jika inputnya berupa vektor (rank-1), maka fungsi ini akan menemukan k entri terbesar dalam vektor dan menghasilkan nilai serta indeksnya sebagai vektor. Dengan demikian, values[j] adalah entri terbesar ke- j dalam input , dan indeksnya adalah indices[j] .

Untuk matriks (atau input peringkat lebih tinggi), menghitung k entri teratas di setiap baris (atau 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:

• Cakupan: Objek Cakupan
• Masukan: 1-D atau lebih tinggi dengan dimensi terakhir minimal k .
• k: 0-D. Jumlah elemen teratas yang dicari di sepanjang dimensi terakhir (di sepanjang setiap baris untuk matriks).

Atribut opsional (lihat Attrs ):

• sorted: Jika bernilai true, maka k elemen yang dihasilkan akan diurutkan berdasarkan nilai-nilainya dalam urutan menurun.

Mengembalikan:

• Nilai Output : k elemen terbesar di sepanjang setiap irisan dimensi terakhir.
• Indeks Output : Indeks values dalam dimensi terakhir dari 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
)