ApproxTopK

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>
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.
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

org.tensorflow.op.PrimitiveOp
booleano final
igual a (objeto obj)
final int
código de hash ()
Operação
op ()
Retorna a Operation subjacente
String final
toString ()
boleano
igual a (objeto arg0)
Aula final<?>
getClass ()
int
código de hash ()
vazio final
notifique ()
vazio final
notifiqueTodos ()
Corda
toString ()
vazio final
espere (long arg0, int arg1)
vazio final
espere (arg0 longo)
vazio final
espere ()

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.