ML Topluluk Günü 9 Kasım! TensorFlow, JAX güncellemeler için bize katılın ve daha fazla bilgi edinin

tensorflow :: ops :: TopK

#include <nn_ops.h>

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

Özet

Giriş bir vektörse (sıra-1), vektördeki en büyük k girişini 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 elde edilen k elemanlar 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 .

Yapıcılar ve Yıkıcılar

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)

Genel özellikler

indices
operation
values

Genel statik işlevler

Sorted (bool x)

Yapılar

tensorflow :: ops :: TopK :: Attrs

TopK için isteğe bağlı öznitelik belirleyiciler.

Genel özellikler

endeksler

::tensorflow::Output indices

operasyon

Operation operation

değerler

::tensorflow::Output values

Kamusal işlevler

TopK

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

TopK

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

Genel statik işlevler

Sıralandı

Attrs Sorted(
  bool x
)