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.