ReadVariableXlaSplitND

ReadVariableXlaSplitND classe final pública

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.

Esta operação pode ser gerada por meio 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]
 
a` outputs` esperado é:
[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]
 

Classes aninhadas

classe ReadVariableXlaSplitND.Options Atributos opcionais para ReadVariableXlaSplitND

Métodos Públicos

estática <T> ReadVariableXlaSplitND <T>
criar ( Scope escopo, Operando <?> recursos, Class <T> T, Long N, List <longo> numSplits, Options ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação ReadVariableXlaSplitND.
Iterator < Operando <T >>
List < Output <T >>
estáticos ReadVariableXlaSplitND.Options
paddings (List <longo> paddings)

Métodos herdados

Métodos Públicos

public static ReadVariableXlaSplitND <T> create ( Scope escopo, Operando <?> recursos, Class <T> T, Long N, List <longo> numSplits, Options ... Opções)

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: "outputs" 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

pública Iterator < Operando <T >> iterator ()

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

public static ReadVariableXlaSplitND.Options paddings (List <longo> paddings)

Parâmetros
recheios 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 divisível uniformemente.
,
ReadVariableXlaSplitND classe final pública

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.

Esta operação pode ser gerada por meio 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]
 
a` outputs` esperado é:
[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]
 

Classes aninhadas

classe ReadVariableXlaSplitND.Options Atributos opcionais para ReadVariableXlaSplitND

Métodos Públicos

estática <T> ReadVariableXlaSplitND <T>
criar ( Scope escopo, Operando <?> recursos, Class <T> T, Long N, List <longo> numSplits, Options ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação ReadVariableXlaSplitND.
Iterator < Operando <T >>
List < Output <T >>
estáticos ReadVariableXlaSplitND.Options
paddings (List <longo> paddings)

Métodos herdados

Métodos Públicos

public static ReadVariableXlaSplitND <T> create ( Scope escopo, Operando <?> recursos, Class <T> T, Long N, List <longo> numSplits, Options ... Opções)

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: "outputs" 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

pública Iterator < Operando <T >> iterator ()

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

public static ReadVariableXlaSplitND.Options paddings (List <longo> paddings)

Parâmetros
recheios 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 divisível uniformemente.