tensorflow :: ops :: TopK

Son boyut için k en büyük öğenin değerlerini ve dizinlerini bulur.

Özet

Girdi bir vektörse (sıra-1), vektördeki en büyük k girdisini bulur ve değerlerini ve indislerini vektör olarak çıkarır. Bu nedenle values[j] olan j içinde inci büyük giriş input ve dizini olan indices[j] .

Matrisler için (daha yüksek sıralı girdi), her satırdaki ilk k girişini hesaplar (son boyut boyunca vektör). Böylece,

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

İki öğe eşitse, alt dizin öğesi önce görünür.

Argümanlar:

• kapsam: Bir Scope nesnesi
• girdi: 1-D veya daha yüksek, son boyut en az k .
• k: 0-D. Son boyut boyunca aranacak üst öğe sayısı (matrisler için her satır boyunca).

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

• sıralı: True ise, sonuçta ortaya çıkan k öğeleri değerlere göre azalan sırada sıralanır.

İadeler:

• Output değerleri: Her son boyutlu dilim boyunca en büyük k eleman.
• Output indeksleri: indislerinin values son boyutu içinde input .

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
)