XlaSplitND

XlaSplitND classe final pública

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

Um op que divide o tensor de entrada 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 XlaSplitND.Options Atributos opcionais para XlaSplitND

Métodos Públicos

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

Métodos herdados

Métodos Públicos

public static XlaSplitND <T> create ( Scope escopo, Operando <T> de entrada, Long N, List <longo> numSplits, Options ... Opções)

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

Parâmetros
alcance escopo atual
entrada 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 XlaSplitND

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

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

public static XlaSplitND.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.