Devuelve los valores min/max k y sus índices del operando de entrada de forma aproximada.
Consulte https://arxiv.org/abs/2206.14286 para conocer los detalles del algoritmo. Esta operación solo está optimizada en TPU actualmente.
Clases anidadas
clase | ApproxTopK.Opciones | Atributos opcionales para ApproxTopK |
Métodos públicos
ApproxTopK.Opciones estáticas | agregadoToTopk (booleano agregadoToTopk) |
estático <T extiende Número> ApproxTopK <T> | crear ( Alcance alcance, Operando <T> entrada, Long k, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación ApproxTopK. |
Salida <Entero> | índices () Los índices de `valores` a lo largo de `reduction_dimension` del operando `input`. |
ApproxTopK.Opciones estáticas | isMaxK (booleano isMaxK) |
ApproxTopK.Opciones estáticas | recordarTarget (Flotante recordarTarget) |
ApproxTopK.Opciones estáticas | Dimensión de reducción (Dimensión de reducción larga) |
ApproxTopK.Opciones estáticas | reducciónInputSizeOverride (Reducción largaInputSizeOverride) |
Salida <T> | valores () Los valores min/max k a lo largo de `reduction_dimension` del operando `input`. |
Métodos Heredados
Métodos públicos
ApproxTopK.Options público estático addedToTopk ( booleano addedToTopk)
Parámetros
agregadoToTopk | 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 la `k` especificada. |
---|
ApproxTopK público estático <T> crear ( Ámbito de alcance, Operando <T> entrada, Long k, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación ApproxTopK.
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 ApproxTopK
Salida pública <Entero> índices ()
Los índices de `valores` a lo largo de `reduction_dimension` del operando `input`.
público estático ApproxTopK.Options isMaxK (booleano isMaxK)
Parámetros
esMaxK | Cuando es verdadero, calcula max-k; de lo contrario, calcula min-k. |
---|
public static ApproxTopK.Options replaceTarget (objetivo de recuperación flotante)
Parámetros
recordarObjetivo | Recuperar el objetivo para la aproximación. Rango en (0,1] |
---|
public static ApproxTopK.OptionsductionDimension (Dimensión de reducción larga)
Parámetros
reducciónDimensión | Dimensión entera a lo largo de la cual buscar. Predeterminado: -1. |
---|
ApproxTopK.Options público estático 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 ser diferido por la forma de la 'entrada'. |
---|
Salida pública <T> valores ()
Los valores min/max k a lo largo de `reduction_dimension` del operando `input`. Las dimensiones son las mismas 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 igual a `k` donde el tamaño está definido por la implementación.