Gather

classe final pública reunir

Reúna fatias do eixo `params` `axis` de acordo com os `índices`.

`indices` deve ser um tensor inteiro de qualquer dimensão (geralmente 0-D ou 1-D). Produz um tensor de saída com forma `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]` onde:

# Scalar indices (output is rank(params) - 1).
     output[a_0, ..., a_n, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices, b_0, ..., b_n]
 
     # Vector indices (output is rank(params)).
     output[a_0, ..., a_n, i, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
 
     # Higher rank indices (output is rank(params) + rank(indices) - 1).
     output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
       params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
 

Observe que na CPU, se um índice fora do limite for encontrado, um erro será retornado. Na GPU, se um índice fora do limite for encontrado, um 0 será armazenado no valor de saída correspondente.

Veja também `tf.batch_gather` e tf.gather_nd .

Classes aninhadas

classe Reunir.Opções Atributos opcionais para Gather

Constantes

Corda OP_NAME O nome desta operação, como conhecido pelo mecanismo principal do TensorFlow

Métodos públicos

Saída <T>
comoSaída ()
Retorna o identificador simbólico do tensor.
static Gather.Options
batchDims (Longo batchDims)
estático <T estende TType > Reúne <T>
create ( Escopo do escopo, Operando <T> params, Operando <? estende TNumber > índices, Operando <? estende TNumber > eixo, Opções... opções)
Método de fábrica para criar uma classe envolvendo uma nova operação Gather.
Saída <T>
saída ()
Valores de `params` reunidos de índices fornecidos por `indices`, com a forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.

Métodos Herdados

Constantes

String final estática pública OP_NAME

O nome desta operação, como conhecido pelo mecanismo principal do TensorFlow

Valor constante: "GatherV2"

Métodos públicos

Public Output <T> asOutput ()

Retorna o identificador simbólico do tensor.

As entradas para operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.

public static Gather.Options batchDims (Longo batchDims)

public static Gather <T> create ( Escopo do escopo, Operando <T> params, Operando <? estende TNumber > índices, Operando <? estende TNumber > eixo, Opções... opções)

Método de fábrica para criar uma classe envolvendo uma nova operação Gather.

Parâmetros
alcance escopo atual
parâmetros O tensor do qual coletar valores. Deve ter pelo menos a classificação `axis + 1`.
índices Tensor de índice. Deve estar no intervalo `[0, params.shape[axis])`.
eixo O eixo em `params` para coletar `índices`. Padrões para a primeira dimensão. Suporta índices negativos.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância do Gather

saída pública <T> saída ()

Valores de `params` reunidos de índices fornecidos por `indices`, com a forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.