Concats el tensor de entrada en todas las dimensiones.
Una operación que fusiona rebanadas del tensor de entrada según el atributo num_splits dado, elimina los rellenos opcionalmente y escribe el tensor fusionado sin rellenos en la variable de recurso.
Esta operación se puede generar a través del puente de TPU.
Por ejemplo, con `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 esperada` outputs` es: [[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
Clases anidadas
clase | AssignVariableXlaConcatND.Options | Los atributos opcionales para AssignVariableXlaConcatND |
Métodos públicos
estática <T> AssignVariableXlaConcatND | crear ( Alcance alcance, operando <?> recursos, Iterable < operando <T >> entradas, List <Larga> numConcats, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación AssignVariableXlaConcatND. |
estáticas AssignVariableXlaConcatND.Options | los acolchados (List <Larga> rellenos) |
Métodos heredados
Métodos públicos
public static AssignVariableXlaConcatND crear ( Alcance alcance, operando <?> recursos, Iterable < operando <T >> entradas, List <Larga> numConcats, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva 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: descripción de "entradas": < |
numConcats | Número de formas de fusionar por dimensión. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de AssignVariableXlaConcatND
públicas estáticas AssignVariableXlaConcatND.Options los acolchados (List <Larga> rellenos)
Parámetros
acolchados | Lista opcional de rellenos derechos por dimensión para eliminar del tensor combinado final. Estos acolchados no deben exceder el tamaño de dimensión del resultado combinado antes de quitar los acolchados. |
---|