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
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. |
---|