Gather

aula final pública Reunir

Reúna fatias do eixo `params` `axis` de acordo com `indices`.

`índices` 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

aula Reunir.Opções Atributos opcionais para Gather

Métodos Públicos

Saída <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
Gather.Options estático
batchDims (batchDims longos)
estático <T, U estende o número, V estende o número> Reunir <T>
criar (escopo do escopo , parâmetros do operando <T>, índices do operando <U>, eixo do operando <V>, opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação Gather.
Saída <T>
saída ()
Valores de `params` coletados de índices dados por `indices`, com forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.

Métodos herdados

Métodos Públicos

Saída pública <T> asOutput ()

Retorna o identificador simbólico de um 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.

público estático Gather.Options batchDims (Longo batchDims)

public static Gather <T> create ( escopo do escopo , parâmetros do operando <T>, índices do operando <U>, eixo do operando <V>, opções... opções)

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

Parâmetros
escopo 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`. O padrão é a primeira dimensão. Suporta índices negativos.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de Gather

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

Valores de `params` coletados de índices dados por `indices`, com forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.