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.
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:]`.