tensor akışı:: işlem:: TopK

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

Özet

Giriş bir vektör (rank-1) ise, vektördeki en büyük k girişi bulur ve bunların değerlerini ve endekslerini vektör olarak çıkarır. Dolayısıyla, values[j] input j en büyük girdidir ve dizini indices[j] 'dir.

Matrisler için (sırasıyla daha yüksek dereceli giriş), her satırdaki en üstteki k girdiyi (son boyut boyunca sırasıyla vektör) hesaplar. Böylece,

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

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

Argümanlar:

• kapsam: Bir Kapsam nesnesi
• giriş: son boyutu en az k olan 1-D veya daha yüksek.
• k: 0-D. Son boyut boyunca (matrisler için her satır boyunca) aranacak üst öğelerin sayısı.

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

• sorted: Eğer doğruysa, sonuçta elde edilen k öğeler, değerlere göre azalan düzende sıralanacaktır.

İade:

• Output değerleri: Her son boyutlu dilim boyunca en büyük k öğe.
• Output endeksleri: input son boyutundaki values endeksleri.

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
)