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]
![](https://www.tensorflow.org/images/Gather.png?hl=pt-br)
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> | |
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
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 <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:]`.