classe final pública AssignVariableXlaConcatND
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 grava o tensor mesclado sem preenchimentos na variável de recurso.
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]]
[2, 2]
[1, 1]
[[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 | |
| static AssignVariableXlaConcatND.Options | preenchimentos (List<Long> preenchimentos) |
Métodos Herdados
Métodos públicos
public static AssignVariableXlaConcatND create ( Escopo do escopo, Operando <?> recurso, Iterable< Operando <T>> entradas, List<Long> numConcats, Opções... 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 { name: "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ático público ( 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. |
|---|