XlaSplitND

aula final pública XlaSplitND

Divide o tensor de entrada em todas as dimensões.

Uma operação que divide o tensor de entrada com base no atributo num_splits fornecido, preenche as fatias opcionalmente e retorna as fatias. As fatias são retornadas em ordem de linha principal.

Esta operação pode ser gerada através da ponte TPU.

Por exemplo, com tensor `input`:

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

`num_splits`:

[2, 2]
 

e `paddings`:

[1, 1]
 

as `saídas` esperadas são:

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

Classes aninhadas

aula

XlaSplitND.Opções

Atributos opcionais para XlaSplitND

Métodos Públicos

estático <T> XlaSplitND <T>	create ( Escopo , Operando <T> entrada, Long N, List<Long> numSplits, Options... opções) Método de fábrica para criar uma classe que envolve uma nova operação XlaSplitND.
Iterador< Operando <T>>	iterador ()
Lista< Saída <T>>	saídas ()
XlaSplitND.Options estático	preenchimentos (List<Long> preenchimentos)

Métodos herdados

Da classe org.tensorflow.op.PrimitiveOp

booleano final	é igual (objeto obj)
int final	código hash ()
Operação	op () Retorna a Operation subjacente
sequência final	para sequenciar ()

Da classe java.lang.Object

boleano	é igual (objeto arg0)
aula final<?>	getClass ()
interno	código hash ()
vazio final	notificar ()
vazio final	notificar todos ()
Corda	para sequenciar ()
vazio final	espere (long arg0, int arg1)
vazio final	espere (arg0 longo)
vazio final	espere ()

Da interface java.lang.Iterable

vazio	forEach (Consumidor<? super T> arg0)
Iterador abstrato< Operando <T>>	iterador ()
Divisor< Operando <T>>	divisor ()

Métodos Públicos

public static XlaSplitND <T> create ( Escopo escopo , entrada Operando <T>, N longo, Lista<Long> numSplits, Opções... opções)

Método de fábrica para criar uma classe que envolve uma nova operação XlaSplitND.

Parâmetros

escopo	escopo atual
entrada	Tensor de entrada para dividir em todas as dimensões. } out_arg { nome: "saídas" descrição: <
numSplits	Número de maneiras de dividir por dimensão. As dimensões da forma devem ser divisíveis igualmente.
opções	carrega valores de atributos opcionais

Devoluções

• uma nova instância do XlaSplitND

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

Lista pública< Saída <T>> saídas ()

preenchimentos XlaSplitND.Options estáticos públicos (preenchimentos List<Long>)

Parâmetros

preenchimentos	Lista opcional de preenchimentos corretos por dimensão do tensor de entrada a ser aplicado antes da divisão. Isso pode ser usado para tornar uma dimensão igualmente divisível.