ApproxTopK

clase final pública ApproxTopK

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.