AssignVariableXlaConcatND

AssignVariableXlaConcatND 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 grava o tensor mesclado sem preenchimentos na variável de recurso.

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 AssignVariableXlaConcatND.Options Atributos opcionais para AssignVariableXlaConcatND

Métodos Públicos

estática <T> AssignVariableXlaConcatND
criar ( Scope escopo, Operando <?> recursos, Iterable < Operando <T >> entradas, List <longo> numConcats, Options ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação AssignVariableXlaConcatND.
estáticos AssignVariableXlaConcatND.Options
paddings (List <longo> paddings)

Métodos herdados

Métodos Públicos

public static AssignVariableXlaConcatND criar ( Scope escopo, Operando <?> recursos, Iterable < Operando <T >> entradas, List <longo> numConcats, Options ... Opções)

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

Parâmetros
alcance escopo atual
recurso Variável de recurso para tensores de entrada concatenados em todas as dimensões. } in_arg {nome: "entradas" 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 AssignVariableXlaConcatND

public static AssignVariableXlaConcatND.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.