Conozca lo último en aprendizaje automático, IA generativa y más en el Simposio WiML 2023.

Se usó la API de Cloud Translation para traducir esta página.

ReadVariableXlaSplitND

clase final pública ReadVariableXlaSplitND

Divide el tensor de entrada de variables de recursos en todas las dimensiones.

Una operación que divide el tensor de entrada de la variable de recursos en función del atributo num_splits dado, rellena los segmentos opcionalmente y devuelve los segmentos. Los sectores se devuelven en orden de fila principal.

Esta operación puede generarse a través del puente de TPU.

Por ejemplo, con el tensor `input`:

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]

`num_splits`:

[2, 2]

y `paddings`:

[1, 1]

la `salida` esperada es:

[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]

Clases anidadas

clase ReadVariableXlaSplitND.Opciones Atributos opcionales para ReadVariableXlaSplitND

Métodos públicos

<T> estático ReadVariableXlaSplitND <T>	crear ( Ámbito ámbito, 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.
Iterador< Operando <T>>	iterador ()
Lista< Salida <T>>	salidas ()
estática ReadVariableXlaSplitND.Options	rellenos (Lista <Long> rellenos)

Métodos Heredados

De la clase org.tensorflow.op.PrimitiveOp

booleano final	es igual a (Objeto obj)
int final	código hash ()
Operación	op () Devuelve la `Operation` subyacente
cadena final	a la cadena ()

De la clase java.lang.Object

booleano	es igual a (Objeto arg0)
clase final<?>	obtenerClase ()
En t	código hash ()
vacío final	notificar ()
vacío final	notificar a todos ()
Cuerda	a la cadena ()
vacío final	espera (largo arg0, int arg1)
vacío final	espera (largo arg0)
vacío final	espera ()

Desde la interfaz java.lang.Iterable

vacío	forEach (Consumidor<? super T> arg0)
Iterador abstracto< Operando <T>>	iterador ()
Spliterator< Operando <T>>	divisor ()

Métodos públicos

public static ReadVariableXlaSplitND <T> create ( Scope scope, Operand <?> resource, Class<T> T, Long N, List<Long> numSplits, Options... options)

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: <
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

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

public List< Salida <T>> salidas ()

rellenos ReadVariableXlaSplitND.Options estáticos públicos ( rellenos List<Long>)

Parámetros

rellenos	Lista opcional de rellenos correctos por dimensión del tensor de entrada para aplicar antes de dividir. Esto se puede usar para hacer una dimensión divisible uniformemente.