TopK

TopK classe final pública

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>
criar ( 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.
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

org.tensorflow.op.RawOp
final booleano
é igual a (obj Object)
final int
Operação
op ()
Retornar esta unidade de computação como uma única Operation .
corda final
boleano
equals (arg0 Object)
aula final <?>
getClass ()
int
hashCode ()
vazio final
notificar ()
vazio final
notifyAll ()
Corda
toString ()
vazio final
wait (longa arg0, int arg1)
vazio final
wait (arg0 de comprimento)
vazio final
wait ()
org.tensorflow.op.Op
abstrato ExecutionEnvironment
env ()
Retorne o ambiente de execução em que este op foi criado.
abstrato Operação
op ()
Retornar esta unidade de computação como uma única Operation .

Constantes

nome_op final String public static

O nome desta operação, conforme conhecido pelo motor principal TensorFlow

Valor constante: "TopKV2"

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.

pública Output <T> values ()

Os maiores elementos `k` ao longo de cada última fatia dimensional.