Retorna valores mínimo/máximo de k e seus índices do operando de entrada de maneira aproximada.
Consulte https://arxiv.org/abs/2206.14286 para obter detalhes do algoritmo. Esta operação está otimizada apenas em TPU atualmente.
Classes aninhadas
aula | AproxTopK.Options | Atributos opcionais para ApproxTopK |
Métodos Públicos
AproxTopK.Options estático | agregadoToTopk (booleano agregadoToTopk) |
estático <T estende número> AproxTopK <T> | |
Saída <Inteiro> | índices () Os índices de `valores` ao longo da `reduction_dimension` do operando `input`. |
AproxTopK.Options estático | isMaxK (booleano isMaxK) |
AproxTopK.Options estático | recallTarget (float recallTarget) |
AproxTopK.Options estático | reduçãoDimension (redução longaDimension) |
AproxTopK.Options estático | 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
público estático approxTopK.Options agregadoToTopk (booleano agregadoToTopk)
Parâmetros
agregadoToTopk | Quando verdadeiro, agrega resultados aproximados ao top-k. Quando falso, retorna os resultados aproximados. O número de resultados aproximados é definido pela implementação e é maior que o `k` especificado. |
---|
public static approxTopK <T> create (escopo do escopo , entrada do operando <T>, k longo, opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação approxTopK.
Parâmetros
escopo | 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 approxTopK
índices <inteiros> de saída pública ()
Os índices de `valores` ao longo da `reduction_dimension` do operando `input`.
público estático approxTopK.Options isMaxK (booleano 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
recordarTarget | Lembre-se do alvo para a aproximação. Intervalo em (0,1] |
---|
público estático AproxTopK.Options reduçãoDimension (redução longaDimension)
Parâmetros
reduçãoDimensão | Dimensão inteira ao longo da qual pesquisar. Padrão: -1. |
---|
público estático approxTopK.Options reduçãoInputSizeOverride (redução longaInputSizeOverride)
Parâmetros
reduçãoInputSizeOverride | Quando definido com um valor positivo, 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 verdadeiro tamanho da entrada não pode ser adiado pelo formato 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 `agregate_to_topk` é verdadeiro, a dimensão de redução é `k`; caso contrário, é maior que `k` onde o tamanho é definido pela implementação.