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 devuelve el tensor fusionado sin rellenos.
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 | XlaConcatND.Opciones | Atributos opcionales para XlaConcatND |
Métodos públicos
Salida <T> | como salida () Devuelve el identificador simbólico de un tensor. |
estático <T> XlaConcatND <T> | crear (alcance alcance , Iterable< Operando <T>> entradas, Lista<Long> numConcats, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación de XlaConcatND. |
Salida <T> | producción () |
XlaConcatND.Opciones estáticas | rellenos (Lista<Largo> rellenos) |
Métodos heredados
Métodos públicos
Salida pública <T> como Salida ()
Devuelve el identificador simbólico de un tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
XlaConcatND <T> estática pública crear (alcance alcance , entradas Iterable < operando <T>>, lista <largo> numConcats, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación de XlaConcatND.
Parámetros
alcance | alcance actual |
---|---|
entradas | Ingrese cortes tensoriales en el orden de la fila principal para fusionarlos en todas las dimensiones. Todas las entradas deben tener la misma forma. } out_arg { nombre: "salida" descripción: < |
númeroConcats | Número de formas de fusionar por dimensión. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de XlaConcatND
Rellenos públicos estáticos XlaConcatND.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. |
---|