Encuentra valores e índices de los elementos más grandes `k` para la última dimensión.
Si la entrada es un vector (rango 1), busca las `k` entradas más grandes en el vector y genera sus valores e índices como vectores. Por lo tanto, `valores [j]` es la `j`-ésima entrada más grande en `entrada`, y su índice es `índices [j]`.
Para matrices (respectivamente, entrada de rango superior), calcula las entradas `k` superiores en cada fila (respectivamente, vector a lo largo de la última dimensión). De este modo,
valores.forma = índices.forma = entrada.forma[:-1] + [k]
Si dos elementos son iguales, el elemento de menor índice aparece primero.
Clases anidadas
| clase | Opciones TopK | Atributos opcionales para TopK | |
Constantes
| Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
| estático <T extiende TNumber > TopK <T> | crear (alcance alcance , entrada del operando <T>, operando <TInt32> k, opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación TopK. |
| Salida < TInt32 > | índices () Los índices de "valores" dentro de la última dimensión de "entrada". |
| Opciones estáticas de TopK | ordenado (ordenado booleano) |
| Salida <T> | valores () Los elementos `k` más grandes a lo largo de cada último segmento dimensional. |
Métodos heredados
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
creación pública estática de TopK <T> (alcance del alcance , entrada del operando <T>, operando <TInt32> k, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación 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
Salida pública < TInt32 > índices ()
Los índices de "valores" dentro de la última dimensión de "entrada".
TopK.Options estático público ordenado (ordenado booleano)
Parámetros
| ordenado | Si es verdadero, los elementos `k` resultantes se ordenarán por valores en orden descendente. |
|---|
valores públicos de salida <T> ()
Los elementos `k` más grandes a lo largo de cada último segmento dimensional.