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

#include <nn_ops.h>

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.

İadeler:

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

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:: Öznitelikler

TopK için isteğe bağlı özellik ayarlayıcılar.

Genel özellikler

endeksler

::tensorflow::Output indices

operasyon

Operation operation

değerler

::tensorflow::Output values

Kamu işlevleri

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ıralanmış

Attrs Sorted(
  bool x
)