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 na ordem da 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_divisões`:
[2, 2]
e `preenchimentos`:
[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

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.