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ükk
eleman. -
Output
indeksleri: indislerininvalues
son boyutu içindeinput
.
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 )