TopK

TopK public final class

Encuentra valores e índices de los elementos más grandes de `k` para la última dimensión.

Si la entrada es un vector (rango-1), encuentra las entradas más grandes `k` en el vector y genera sus valores e índices como vectores. Así, `valores [j]` es la `j` -ésima entrada más grande en` entrada`, y su índice es `índices [j]`.

Para matrices (resp. Entrada de rango superior), calcula las entradas `k` superiores en cada fila (resp. Vector a lo largo de la última dimensión). Por lo tanto,

valores.forma = índices.forma = entrada.forma [: - 1] + [k]

Si dos elementos son iguales, el elemento de índice más bajo aparece primero.

Clases anidadas

clase TopK.Options Los atributos opcionales para TopK

Constantes

Cuerda OP_NAME El nombre de esta operación, como lo conoce el motor central de TensorFlow

Métodos públicos

estática <T se extiende TNumber > TopK <T>
crear ( Alcance alcance, operando <T> de entrada, operando < TInt32 > k, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación de TopK.
Salida < TInt32 >
índices ()
Los índices de "valores" dentro de la última dimensión de "entrada".
estáticas TopK.Options
ordenados (booleano ordenadas)
Salida <T>
valores ()
Los elementos más grandes de `k` a lo largo de cada último corte dimensional.

Métodos heredados

Constantes

OP_NAME pública final static String

El nombre de esta operación, como lo conoce el motor central de TensorFlow

Valor constante: "TopKV2"

Métodos públicos

public static TopK <T> crear ( Alcance alcance, operando <T> de entrada, operando < TInt32 > k, Opciones ... Opciones)

Método de fábrica para crear una clase que envuelva una nueva operación de TopK.

Parámetros
alcance alcance actual
aporte 1-D o superior con la última dimensión al menos "k".
k 0-D. Número de elementos superiores a buscar a lo largo de la última dimensión (a lo largo de cada fila para matrices).
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de TopK

pública de salida < TInt32 > índices ()

Los índices de "valores" dentro de la última dimensión de "entrada".

estáticas pública TopK.Options ordenados (booleano ordenadas)

Parámetros
ordenado Si es verdadero, los elementos `k` resultantes serán ordenados por los valores en orden descendente.

pública de salida <T> valores de ()

Los elementos más grandes de `k` a lo largo de cada último corte dimensional.