aula final pública AproxTopK
Retorna valores min/max k e seus índices do operando de entrada de forma aproximada.
Consulte https://arxiv.org/abs/2206.14286 para obter os detalhes do algoritmo. Este op é otimizado apenas em TPU atualmente.
Classes aninhadas
| classe | ApproxTopK.Options | Atributos opcionais para ApproxTopK | |
Métodos públicos
| estático ApproxTopK.Options | agregadoToTopk (booleano agregadoToTopk) |
| estático <T estende Número> ApproxTopK <T> | |
| Saída <Inteiro> | índices () Os índices de `values` ao longo da `reduction_dimension` do operando `input`. |
| estático ApproxTopK.Options | isMaxK (Booleano isMaxK) |
| estático ApproxTopK.Options | recallTarget (Float recallTarget) |
| estático ApproxTopK.Options | reduçãoDimensão (Longa reduçãoDimensão) |
| estático ApproxTopK.Options | reduçãoInputSizeOverride (Redução longaInputSizeOverride) |
| Saída <T> | valores () Os valores min/max k ao longo da `reduction_dimension` do operando `input`. |
Métodos Herdados
Métodos públicos
public static ApproxTopK.Options agregadoToTopk (Booleano agregadoToTopk)
Parâmetros
| agregadoParaTopk | Quando true, agrega resultados aproximados ao top-k. Quando false, retorna os resultados aproximados. O número de resultados aproximados é definido pela implementação e é maior igual ao `k` especificado. |
|---|
public static ApproxTopK <T> create ( Escopo do escopo, entrada Operando <T>, Long k, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação ApprTopK.
Parâmetros
| alcance | escopo atual |
|---|---|
| entrada | Matriz para pesquisar. Deve ser pelo menos 1-D do tipo flutuante |
| k | Especifica o número de min/max-k. |
| opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de ApprTopK
índices de saída pública <Integer> ()
Os índices de `values` ao longo da `reduction_dimension` do operando `input`.
public static ApproxTopK.Options isMaxK (Boolean isMaxK)
Parâmetros
| isMaxK | Quando verdadeiro, calcula max-k; caso contrário, calcula min-k. |
|---|
público estático ApproxTopK.Options recallTarget (Float recallTarget)
Parâmetros
| recallTarget | Recall alvo para a aproximação. Intervalo em (0,1] |
|---|
public static ApproxTopK.Options reduceDimension ( Long reduceDimension)
Parâmetros
| reduçãoDimensão | Dimensão inteira ao longo da qual pesquisar. Padrão: -1. |
|---|
public static ApproxTopK.Options reduçãoInputSizeOverride ( longo reduçãoInputSizeOverride)
Parâmetros
| reduçãoInputSizeOverride | Quando definido como um valor positivo, ele substitui o tamanho determinado por `input[reduction_dim]` para avaliar o recall. Esta opção é útil quando a `entrada` fornecida é apenas um subconjunto da computação geral em SPMD ou pipelines distribuídos, onde o tamanho real da entrada não pode ser adiado pela forma da `entrada`. |
|---|
valores de saída pública <T> ()
Os valores min/max k ao longo da `reduction_dimension` do operando `input`. A dimensão é a mesma do operando `input` exceto para `reduction_dimension`: quando `aggregate_to_topk` for true, a dimensão de redução será `k`; caso contrário, é maior igual a `k` onde o tamanho é definido pela implementação.