XlaConcatND

XlaConcatND classe final pública

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

Um op 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 por meio 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]
a` outputs` esperado é: 
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]

Classes aninhadas

classe XlaConcatND.Options Atributos opcionais para XlaConcatND

Métodos Públicos

Output <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
estática <T> XlaConcatND <T>
criar ( Scope escopo, Iterable < Operando <T >> entradas, List <longo> numConcats, Options ... Opções)
Método de fábrica para criar uma classe envolvendo uma nova operação XlaConcatND.
Output <T>
saída ()
estáticos XlaConcatND.Options
paddings (List <longo> paddings)

Métodos herdados

Métodos Públicos

pública Output <T> asOutput ()

Retorna o identificador simbólico de um tensor.

As entradas para as 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 ( Scope escopo, Iterable < Operando <T >> entradas, List <longo> numConcats, Options ... Opções)

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

Parâmetros
alcance escopo atual
entradas Fatias de tensor de entrada na ordem maior da linha para mesclar em todas as dimensões. Todas as entradas devem ter o mesmo formato. } out_arg {nome: descrição de "saída": <
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

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

public static XlaConcatND.Options paddings (List <longo> paddings)

Parâmetros
recheios 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 de remover os preenchimentos.