ReadVariableXlaSplitND

classe final pública ReadVariableXlaSplitND

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

Um op 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 principal da linha.

Este op pode ser gerado através da ponte TPU.

Por exemplo, com o 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 do escopo, Operando <?> recurso, Class<T> T, Long N, List<Long> numSplits, Options... options)
Método de fábrica para criar uma classe que envolve uma nova operação ReadVariableXlaSplitND.
Iterador< Operando <T>>
iterador ()
Lista< Saída <T>>
saídas ()
static ReadVariableXlaSplitND.Options
preenchimentos (List<Long> preenchimentos)

Métodos Herdados

Métodos públicos

public static ReadVariableXlaSplitND <T> create ( Escopo do escopo, Operando <?> recurso, Class<T> T, Long N, List<Long> numSplits, Options... options)

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

Parâmetros
alcance escopo atual
recurso Variável de recurso do tensor de entrada para dividir em todas as dimensões. } out_arg { name: "saídas" descrição: <
numSplits Número de maneiras de dividir por dimensão. As dimensões da forma devem ser divisíveis uniformemente.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de ReadVariableXlaSplitND

public Iterator< Operando <T>> iterador ()

public List< Output <T>> outputs ()

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

Parâmetros
acolchoamentos Lista opcional de preenchimentos corretos por dimensão do tensor de entrada para aplicar antes da divisão. Isso pode ser usado para tornar uma dimensão uniformemente divisível.