tensorflow :: ops :: TopK

#include <nn_ops.h>

Menemukan nilai dan indeks dari k elemen terbesar untuk dimensi terakhir.

Ringkasan

Jika input adalah vektor (peringkat-1), temukan entri terbesar k dalam vektor dan output nilai dan indeksnya sebagai vektor. Jadi values[j] adalah entri terbesar ke- j dalam input , dan indeksnya adalah indices[j] .

Untuk matriks (masing-masing masukan peringkat yang lebih tinggi), hitung entri k teratas di setiap baris (tanggapan vektor sepanjang dimensi terakhir). Jadi, 

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

Jika dua elemen sama, elemen indeks lebih rendah muncul pertama.

Argumen:

  • scope: Objek Scope
  • masukan: 1-D atau lebih tinggi dengan dimensi terakhir setidaknya k .
  • k: 0-D. Jumlah elemen teratas yang harus dicari di sepanjang dimensi terakhir (di sepanjang setiap baris untuk matriks).

Atribut opsional (lihat Attrs ):

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

Pengembalian:

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

Pembuat dan Penghancur

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)

Structs

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

Disortir

Attrs Sorted(
  bool x
)