Divide el tensor de entrada de la 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 segmentos de forma opcional y devuelve los segmentos. Los cortes se devuelven en orden de fila principal.
Esta operación se puede generar a través del puente de TPU.
Por ejemplo, con `tensor input`:
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
` num_splits`: [2, 2]
y `paddings`: [1, 1]
la esperada` outputs` es: [[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Clases anidadas
clase | ReadVariableXlaSplitND.Options | Los atributos opcionales para ReadVariableXlaSplitND |
Métodos públicos
estática <T> ReadVariableXlaSplitND <T> | crear ( Alcance alcance, operando <?> Recursos, Clase <T> T, Long N, List <Larga> numSplits, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación ReadVariableXlaSplitND. |
Iterador < operando <T >> | iterador () |
Lista < salida <T >> | salidas () |
estáticas ReadVariableXlaSplitND.Options | los acolchados (List <Larga> rellenos) |
Métodos heredados
Métodos públicos
public static ReadVariableXlaSplitND <T> crear ( Alcance alcance, operando <?> Recursos, Clase <T> T, Long N, List <Larga> numSplits, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva 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: < |
numSplits | 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
públicas estáticas ReadVariableXlaSplitND.Options los acolchados (List <Larga> rellenos)
Parámetros
acolchados | Lista opcional de rellenos derechos por dimensión del tensor de entrada para aplicar antes de dividir. Esto se puede usar para hacer que una dimensión sea divisible de manera uniforme. |
---|