aliran tensor:: operasi:: TopK

#include <nn_ops.h>

Menemukan nilai dan indeks k elemen terbesar untuk dimensi terakhir.

Ringkasan

Jika masukannya adalah vektor (peringkat-1), carilah k entri terbesar dalam vektor tersebut dan keluarkan nilai dan indeksnya sebagai vektor. Jadi values[j] adalah entri terbesar ke- j di input , dan indeksnya adalah indices[j] .

Untuk matriks (masing-masing masukan dengan peringkat lebih tinggi), hitung k entri teratas di setiap baris (masing-masing vektor sepanjang dimensi terakhir). Dengan demikian,

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

Jika dua elemen sama, elemen dengan indeks lebih rendah akan muncul terlebih dahulu.

Argumen:

  • ruang lingkup: Objek Lingkup
  • input: 1-D atau lebih tinggi dengan dimensi terakhir setidaknya k .
  • k: 0-D. Jumlah elemen teratas yang harus dicari sepanjang dimensi terakhir (sepanjang setiap baris matriks).

Atribut opsional (lihat Attrs ):

  • diurutkan: Jika benar, k elemen yang dihasilkan akan diurutkan berdasarkan nilai dalam urutan menurun.

Pengembalian:

  • Nilai Output : k elemen terbesar sepanjang setiap potongan dimensi terakhir.
  • Indeks Output : Indeks values dalam dimensi terakhir input .

Konstruktor dan Destruktor

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)

Atribut publik

indices
operation
values

Fungsi statis publik

Sorted (bool x)

Struktur

tensorflow:: ops:: TopK:: Attrs

Penyetel atribut opsional untuk TopK .

Atribut publik

indeks 

::tensorflow::Output indices

operasi 

Operation operation

nilai-nilai 

::tensorflow::Output values

Fungsi publik

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
)

Fungsi statis publik

Diurutkan 

Attrs Sorted(
  bool x
)