XlaConcatND

aula final pública XlaConcatND

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

Uma operação que mescla fatias do tensor de entrada com base no atributo num_splits fornecido, remove os preenchimentos opcionalmente e retorna o tensor mesclado sem preenchimentos.

Esta operação pode ser gerada através da ponte TPU.

Por exemplo, com 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.Opções Atributos opcionais para XlaConcatND

Métodos Públicos

Saída <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
estático <T> XlaConcatND <T>
create (escopo do escopo , entradas Iterable< Operand <T>>, List<Long> numConcats, Options... options)
Método de fábrica para criar uma classe que envolve uma nova operação XlaConcatND.
Saída <T>
saída ()
XlaConcatND.Options estático
preenchimentos (List<Long> preenchimentos)

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 XlaConcatND <T> create ( escopo de escopo , entradas Iterable< Operand <T>>, List<Long> numConcats, Options... options)

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

Parâmetros
escopo escopo atual
entradas Insira fatias de tensor na ordem principal da linha para mesclar todas as dimensões. Todas as entradas devem ter o mesmo formato. } out_arg { nome: "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
preenchimentos Lista opcional de preenchimentos direitos por dimensão a serem removidos do tensor mesclado final. Esses preenchimentos não devem exceder o tamanho da dimensão do resultado mesclado antes de remover os preenchimentos.