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

Da classe 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 ()

Da classe java.lang.Object

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.