Tensor de entrada concats en todas las dimensiones.
Una operación que fusiona corta el 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 TPU.
Por ejemplo, con el tensor "entrada":
[[0, 1],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
`num_divisiones`: [2, 2]
y `rellenos`: [1, 1]
los "resultados" esperados son: [[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
Clases anidadas
clase | AsignarVariableXlaConcatND.Options | Atributos opcionales para AssignVariableXlaConcatND |
Métodos públicos
estático <T> AsignarVariableXlaConcatND | crear (alcance alcance , recurso Operando <?>, entradas Iterable< Operando <T>>, 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<Largo> rellenos) |
Métodos heredados
Métodos públicos
creación estática pública AssignVariableXlaConcatND (alcance alcance , recurso Operando <?>, entradas Iterable< Operando <T>>, Lista<Long> numConcats, Opciones... opciones)
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: < |
númeroConcats | 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 de lista <largo>)
Parámetros
acolchados | Lista opcional de rellenos derechos por dimensión para eliminar del tensor fusionado final. Estos rellenos no deben exceder el tamaño de la dimensión del resultado combinado antes de quitar los rellenos. |
---|