ReadVariableXlaSplitND

ReadVariableXlaSplitND public final class

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 >>
Lista < salida <T >>
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ública Iterator < operando <T >> iterador ()

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

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.