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