Concatena el tensor de entrada en todas las dimensiones.
Una operación que fusiona divide el tensor de entrada en función del atributo num_splits dado, elimina los rellenos opcionalmente y escribe el tensor fusionado sin rellenos en la variable de recurso.
Esta operación puede generarse a través del puente de TPU.
Por ejemplo, con el tensor `input`:
[[0, 1],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
`num_splits`: [2, 2]
y `paddings`: [1, 1]
la `salida` esperada es: [[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
Clases anidadas
clase | AsignarVariableXlaConcatND.Opciones | Atributos opcionales para AssignVariableXlaConcatND |
Métodos públicos
<T> estática AsignarVariableXlaConcatND | crear ( Ámbito ámbito, Operando <?> recurso, Iterable< Operando <T>> entradas, Lista<Long> numConcats, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación AssignVariableXlaConcatND. |
estático AssignVariableXlaConcatND.Options | rellenos (Lista <Long> rellenos) |
Métodos Heredados
Métodos públicos
public static AssignVariableXlaConcatND create ( Scope scope, Operand <?> resource, Iterable< Operand <T>> input, List<Long> numConcats, Options... options)
Método de fábrica para crear una clase que envuelve una nueva operación AssignVariableXlaConcatND.
Parámetros
alcance | alcance actual |
---|---|
recurso | Variable de recurso para tensores de entrada concatenados en todas las dimensiones. } in_arg { nombre: "entradas" descripción: < |
numConcats | Número de formas de fusionar por dimensión. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de AssignVariableXlaConcatND
rellenos públicos estáticos AssignVariableXlaConcatND.Options (rellenos List<Long>)
Parámetros
rellenos | Lista opcional de rellenos correctos por dimensión para quitar del tensor fusionado final. Estos rellenos no deben exceder el tamaño de la dimensión del resultado fusionado antes de quitar los rellenos. |
---|