XlaSplitND

clase final pública XlaSplitND

Divide el tensor de entrada en todas las dimensiones.

Una operación que corta el tensor de entrada según el atributo num_splits dado, rellena los cortes opcionalmente y devuelve los cortes. Los sectores se devuelven en orden de fila principal.

Esta operación se puede generar a través del puente TPU.

Por ejemplo, con el tensor `input`:

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
 
`num_splits`:
[2, 2]
 
y `paddings`:
[1, 1]
 
las `salidas` esperadas son:
[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]
 

Clases anidadas

clase XlaSplitND.Opciones Atributos opcionales para XlaSplitND

Métodos públicos

estático <T> XlaSplitND <T>
crear (alcance alcance , entrada de operando <T>, N largo, lista <largo> numSplits, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación XlaSplitND.
Iterador< Operando <T>>
Lista< Salida <T>>
XlaSplitND.Options estático
rellenos (Lista<Largo> rellenos)

Métodos heredados

Métodos públicos

creación pública estática de XlaSplitND <T> (alcance del alcance , entrada del operando <T>, N largo, lista <Long> numSplits, opciones... opciones)

Método de fábrica para crear una clase que envuelve una nueva operación XlaSplitND.

Parámetros
alcance alcance actual
aporte Tensor de entrada para dividir en todas las dimensiones. } out_arg { nombre: "salidas" descripción: <
numDivisiones Número de formas de dividir por dimensión. Las dimensiones de la forma deben ser divisibles uniformemente.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de XlaSplitND

Iterador público< Operando <T>> iterador ()

Lista pública< Salida <T>> salidas ()

Rellenos públicos estáticos XlaSplitND.Options (rellenos de lista <largo>)

Parámetros
acolchados Lista opcional de rellenos derechos por dimensión del tensor de entrada que se aplicarán antes de dividir. Esto se puede utilizar para hacer que una dimensión sea divisible uniformemente.