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.

XlaSplitND

clase final pública XlaSplitND

Divide el tensor de entrada en todas las dimensiones.

Una operación que divide el tensor de entrada 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 XlaSplitND.Opciones Atributos opcionales para XlaSplitND

Métodos públicos

estático <T> XlaSplitND <T>	crear ( Ámbito de ámbito, Operando <T> entrada, Long N, List<Long> numSplits, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación XlaSplitND.
Iterador< Operando <T>>	iterador ()
Lista< Salida <T>>	salidas ()
XlaSplitND estático.Opciones	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	esperar (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 XlaSplitND <T> create ( Scope scope, Operand <T> input, Long N, List<Long> numSplits, Options... options)

Método de fábrica para crear una clase que envuelve una nueva operación XlaSplitND.

Parámetros

alcance	alcance actual
aporte	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 XlaSplitND

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

public List< Salida <T>> salidas ()

rellenos XlaSplitND.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.