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