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 devuelve el tensor fusionado sin rellenos.
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 | 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 ( Ámbito de ámbito, Iterable< Operando <T>> entradas, Lista<Long> numConcats, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación XlaConcatND. |
Salida <T> | salida () |
estático XlaConcatND.Opciones | rellenos (Lista <Long> rellenos) |
Métodos Heredados
Métodos públicos
salida pública <T> como salida ()
Devuelve el identificador simbólico de un tensor.
Las entradas de 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> create ( Scope scope, Iterable< Operand <T>> entradas, List<Long> numConcats, Options... options)
Método de fábrica para crear una clase que envuelve una nueva operación XlaConcatND.
Parámetros
alcance | alcance actual |
---|---|
entradas | Ingrese cortes de tensor en orden de fila principal para fusionar todas las dimensiones. Todas las entradas deben tener la misma forma. } out_arg { nombre: "salida" descripción: < |
numConcats | Número de formas de fusionar por dimensión. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de XlaConcatND
rellenos XlaConcatND.Options estáticos públicos ( 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. |
---|