XlaConcatND

classe final pública XlaConcatND

Concatena o tensor de entrada em todas as dimensões.

Um op que mescla fatias o tensor de entrada com base no atributo num_splits fornecido, remove os preenchimentos opcionalmente e retorna o tensor mesclado sem preenchimentos.

Este op pode ser gerado através da ponte TPU.

Por exemplo, com o tensor `input`:

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
 
`num_splits`:
[2, 2]
 
e `paddings`:
[1, 1]
 
as `saídas` esperadas são:
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]
 

Classes aninhadas

aula XlaConcatND.Options Atributos opcionais para XlaConcatND

Métodos públicos

Saída <T>
comoSaída ()
Retorna o identificador simbólico de um tensor.
estático <T> XlaConcatND <T>
create ( Escopo do escopo, Iterable< Operando <T>> entradas, List<Long> numConcats, Opções... opções)
Método de fábrica para criar uma classe envolvendo uma nova operação XlaConcatND.
Saída <T>
saída ()
static XlaConcatND.Options
preenchimentos (List<Long> preenchimentos)

Métodos Herdados

Métodos públicos

Public Output <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 XlaConcatND <T> create ( Escopo do escopo, Iterable< Operando <T>> entradas, List<Long> numConcats, Opções... opções)

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

Parâmetros
alcance escopo atual
entradas Insira fatias de tensor em ordem de linha principal para mesclar em todas as dimensões. Todas as entradas devem ter a mesma forma. } out_arg { name: "saída" descrição: <
numConcats Número de maneiras de mesclar por dimensão.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de XlaConcatND

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

preenchimentos XlaConcatND.Options estáticos públicos ( preenchimentos List<Long>)

Parâmetros
acolchoamentos Lista opcional de preenchimentos corretos por dimensão para retirar do tensor mesclado final. Esses preenchimentos não devem exceder o tamanho da dimensão do resultado mesclado antes da remoção dos preenchimentos.