Encontra valores e índices dos maiores elementos `k` para a última dimensão.
Se a entrada for um vetor (classificação-1), encontra as `k` maiores entradas no vetor e gera seus valores e índices como vetores. Assim, `values[j]` é a `j`-ésima maior entrada em `input`, e seu índice é `indices[j]`.
Para matrizes (resp. entrada de classificação mais alta), calcula as principais entradas `k` em cada linha (resp. vetor ao longo da última dimensão). Por isso,
valores.forma = índices.forma = entrada.forma[:-1] + [k]
Se dois elementos forem iguais, o elemento de índice inferior aparece primeiro.
Classes aninhadas
aula | TopK.Opções | Atributos opcionais para TopK |
Constantes
Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
estático <T estende TNumber > TopK <T> | |
Saída < TInt32 > | índices () Os índices de `valores` na última dimensão de `entrada`. |
TopK.Options estático | classificado (classificado booleano) |
Saída <T> | valores () Os maiores elementos `k` ao longo de cada última fatia dimensional. |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
public static TopK <T> create ( Escopo escopo , Operando <T> entrada, Operando < TInt32 > k, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação TopK.
Parâmetros
escopo | escopo atual |
---|---|
entrada | 1-D ou superior com última dimensão pelo menos `k`. |
k | 0-D. Número de elementos superiores a procurar ao longo da última dimensão (ao longo de cada linha para matrizes). |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância do TopK
public static TopK.Options classificado (classificado booleano)
Parâmetros
classificado | Se for verdade, os elementos `k` resultantes serão classificados pelos valores em ordem decrescente. |
---|