Divide el tensor de entrada de variable de recurso en todas las dimensiones.
Una operación que divide el tensor de entrada de la variable de recurso en función del atributo num_splits dado, rellena los sectores opcionalmente y devuelve los sectores. 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]]
`num_divisiones`: [2, 2]
y `rellenos`: [1, 1]
los "resultados" esperados son: [[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Clases anidadas
clase | LeerVariableXlaSplitND.Options | Atributos opcionales para ReadVariableXlaSplitND |
Métodos públicos
estático <T> ReadVariableXlaSplitND <T> | crear (alcance alcance , recurso Operando <?>, Clase<T> T, N largo, Lista<Largo> numSplits, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación ReadVariableXlaSplitND. |
Iterador< Operando <T>> | iterador () |
Lista< Salida <T>> | salidas () |
ReadVariableXlaSplitND.Options estático | rellenos (Lista<Largo> rellenos) |
Métodos heredados
Métodos públicos
public static ReadVariableXlaSplitND <T> create (Ámbito de alcance , Operando <?> recurso, Clase<T> T, Long N, List<Long> numSplits, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación ReadVariableXlaSplitND.
Parámetros
alcance | alcance actual |
---|---|
recurso | Variable de recurso del 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 ReadVariableXlaSplitND
rellenos públicos estáticos ReadVariableXlaSplitND.Options (rellenos List<Long>)
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. |
---|