AssignVariableXlaConcatND

classe final pública AssignVariableXlaConcatND

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

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_divisões`:
[2, 2]
e `preenchimentos`:
[1, 1]
as `saídas` esperadas são:
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]

Classes aninhadas

aula AssignVariableXlaConcatND.Options Atributos opcionais para AssignVariableXlaConcatND

Métodos Públicos

estático <T> AssignVariableXlaConcatND
create ( Escopo do escopo, recurso Operando <?>, entradas Iterable< Operando <T>>, Lista<Long> numConcats, Opções... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação AssignVariableXlaConcatND.
AssignVariableXlaConcatND.Options estático
preenchimentos (List<Long> preenchimentos)

Métodos herdados

Métodos Públicos

public static AssignVariableXlaConcatND create ( Escopo de escopo, recurso Operando <?>, entradas Iterable< Operando <T>>, List<Long> numConcats, Opções... opções)

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

Parâmetros
escopo 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

preenchimentos AssignVariableXlaConcatND.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.