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.