tensorflow::ops::TopK

Son boyut için en büyük k elemanın değerlerini ve indekslerini bulur.

Özet

Giriş bir vektör (1. mertebe) ise, vektördeki en büyük k elemanı bulur ve değerlerini ve indekslerini vektör olarak çıktı verir. Dolayısıyla values[j] , input j . en büyük elemandır ve indeksi indices[j] 'dir.

Matrisler için (veya daha yüksek sıralı girdiler için), her satırdaki (veya son boyuttaki vektördeki) en üst k girdiyi hesaplar. Dolayısıyla,

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

İki eleman eşitse, daha düşük indeksli eleman önce görünür.

Argümanlar:

• kapsam: Bir Kapsam nesnesi
• Giriş: Son boyutu en az k olan 1 boyutlu veya daha yüksek boyutlu bir veri seti.
• k: 0-Boyut. Son boyut boyunca (matrisler için her satır boyunca) aranacak en üstteki eleman sayısı.

İsteğe bağlı özellikler (bkz. Attrs ):

• sorted: Eğer true ise, elde edilen k eleman değerlerine göre azalan sırada sıralanacaktır.

İadeler:

• Output değerleri: Her bir son boyutlu dilim boyunca en büyük k eleman.
• Output indeksleri: input son boyutundaki values indeksleri.

Genel özellikler

::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
)