XlaConcatND

clase final pública XlaConcatND

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>
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

Salida pública <T> salida ()

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.