clase final pública AproxTopK
Devuelve los valores min/max k y sus índices del operando de entrada de forma aproximada.
Consulte https://arxiv.org/abs/2206.14286 para obtener detalles del algoritmo. Actualmente, esta operación solo está optimizada en TPU.
Clases anidadas
| clase | Opciones de AproxTopK | Atributos opcionales para ApproxTopK | |
Métodos públicos
| Opciones estáticas de AproxTopK | agregadoToTopk (agregado booleanoToTopk) |
| estático <T extiende Número> AproxTopK <T> | crear (alcance alcance , entrada Operando <T>, k larga, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación AproxTopK. |
| Salida <Entero> | índices () Los índices de "valores" a lo largo de la "dimensión_reducción" del operando "entrada". |
| Opciones estáticas de AproxTopK | isMaxK (booleano isMaxK) |
| Opciones estáticas de AproxTopK | recordarTarget (objetivo de recuperación flotante) |
| Opciones estáticas de AproxTopK | dimensión de reducción (dimensión de reducción larga) |
| Opciones estáticas de AproxTopK | reducciónInputSizeOverride (reducción largaInputSizeOverride) |
| Salida <T> | valores () Los valores mínimo/máximo k a lo largo de la `dimensión_reducción` del operando `entrada`. |
Métodos heredados
Métodos públicos
público estático AproxTopK.Options agregadoToTopk (agregado booleanoToTopk)
Parámetros
| agregadoATopk | Cuando es verdadero, agrega resultados aproximados a top-k. Cuando es falso, devuelve los resultados aproximados. El número de resultados aproximados está definido por la implementación y es mayor que el "k" especificado. |
|---|
creación pública estática de AproxTopK <T> (alcance alcance , entrada de operando <T>, k larga, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación AproxTopK.
Parámetros
| alcance | alcance actual |
|---|---|
| aporte | Matriz para buscar. Debe ser al menos 1-D del tipo flotante. |
| k | Especifica el número de min/max-k. |
| opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de AproxTopK
índices públicos de salida <entero> ()
Los índices de "valores" a lo largo de la "dimensión_reducción" del operando "entrada".
público estático AproxTopK.Options isMaxK (booleano isMaxK)
Parámetros
| esMaxK | Cuando es verdadero, calcula max-k; de lo contrario calcula min-k. |
|---|
público estático AproxTopK.Options recordarTarget (objetivo de recuperación flotante)
Parámetros
| recordarObjetivo | Recuerde el objetivo para la aproximación. Rango en (0,1] |
|---|
dimensión de reducción estática pública AproxTopK.Options (dimensión de reducción larga)
Parámetros
| reducciónDimensión | Dimensión entera a lo largo de la cual buscar. Predeterminado: -1. |
|---|
público estático AproxTopK.Options reducciónInputSizeOverride (reducción largaInputSizeOverride)
Parámetros
| reducciónInputSizeOverride | Cuando se establece en un valor positivo, anula el tamaño determinado por `input[reduction_dim]` para evaluar la recuperación. Esta opción es útil cuando la "entrada" dada es solo un subconjunto del cálculo general en SPMD o canalizaciones distribuidas, donde el tamaño de entrada real no puede diferirse por la forma de la "entrada". |
|---|
valores públicos de salida <T> ()
Los valores mínimo/máximo k a lo largo de la `dimensión_reducción` del operando `entrada`. La dimensión es la misma que el operando `input` excepto por `reduction_dimension`: cuando `aggregate_to_topk` es verdadero, la dimensión de reducción es `k`; de lo contrario, es mayor que "k", donde el tamaño está definido por la implementación.