ReadVariableXlaSplitND

classe final pública ReadVariableXlaSplitND

Divide o tensor de entrada da variável de recurso em todas as dimensões.

Uma operação que divide o tensor de entrada da variável de recurso com base no atributo num_splits fornecido, preenche 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_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 ReadVariableXlaSplitND.Options Atributos opcionais para ReadVariableXlaSplitND

Métodos Públicos

estático <T> ReadVariableXlaSplitND <T>
create ( Escopo , Operando <?> recurso, Class<T> T, Long N, List<Long> numSplits, Options... options)
Método de fábrica para criar uma classe que agrupa uma nova operação ReadVariableXlaSplitND.
Iterador< Operando <T>>
Lista< Saída <T>>
ReadVariableXlaSplitND.Options estático
preenchimentos (List<Long> preenchimentos)

Métodos herdados

Métodos Públicos

public static ReadVariableXlaSplitND <T> create ( Escopo escopo , Operando <?> recurso, Classe<T> T, Long N, List<Long> numSplits, Opções... opções)

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

Parâmetros
escopo escopo atual
recurso Variável de recurso do 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 de ReadVariableXlaSplitND

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

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

preenchimentos públicos estáticos ReadVariableXlaSplitND.Options (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.