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 "entrada":
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
[2, 2]
[1, 1]
[[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 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>> | iterador () |
Lista< Salida <T>> | salidas () |
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
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. |
---|