AssignVariableXlaConcatND

clase final pública AssignVariableXlaConcatND

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 escribe el tensor fusionado sin rellenos en la variable de recurso.

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 AsignarVariableXlaConcatND.Opciones Atributos opcionales para AssignVariableXlaConcatND

Métodos públicos

<T> estática AsignarVariableXlaConcatND
crear ( Ámbito ámbito, Operando <?> recurso, Iterable< Operando <T>> entradas, 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 <Long> rellenos)

Métodos Heredados

Métodos públicos

public static AssignVariableXlaConcatND create ( Scope scope, Operand <?> resource, Iterable< Operand <T>> input, List<Long> numConcats, Options... options)

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: <
numConcats 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 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.