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