ReadVariableXlaSplitND

clase final pública ReadVariableXlaSplitND

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 `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 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>>
Lista< Salida <T>>
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, N largo, Lista<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

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

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

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.