Concats el tensor de entrada 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 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 | XlaConcatND.Options | Los atributos opcionales para XlaConcatND |
Métodos públicos
Salida <T> | asOutput () Devuelve el identificador simbólico de un tensor. |
estática <T> XlaConcatND <T> | crear ( Alcance alcance, Iterable < operando <T >> entradas, List <Larga> numConcats, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación XlaConcatND. |
Salida <T> | salida () |
estáticas XlaConcatND.Options | los acolchados (List <Larga> rellenos) |
Métodos heredados
Métodos públicos
pública de salida <T> asOutput ()
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.
public static XlaConcatND <T> crear ( Alcance alcance, Iterable < operando <T >> entradas, List <Larga> numConcats, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación XlaConcatND.
Parámetros
alcance | alcance actual |
---|---|
entradas | Ingrese segmentos de tensor en orden de fila principal para fusionar todas las dimensiones. Todas las entradas deben tener la misma forma. } out_arg {nombre: descripción de "salida": < |
numConcats | Número de formas de fusionar por dimensión. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de XlaConcatND
públicas estáticas XlaConcatND.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. |
---|