Divide o tensor de entrada em todas as dimensões.
Uma operação que divide o tensor de entrada com base no atributo num_splits fornecido, preenche as 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]]
[2, 2]
[1, 1]
[[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Classes aninhadas
aula | XlaSplitND.Opções | Atributos opcionais para XlaSplitND |
Métodos Públicos
estático <T> XlaSplitND <T> | create ( Escopo , Operando <T> entrada, Long N, List<Long> numSplits, Options... opções) Método de fábrica para criar uma classe que envolve uma nova operação XlaSplitND. |
Iterador< Operando <T>> | iterador () |
Lista< Saída <T>> | saídas () |
XlaSplitND.Options estático | preenchimentos (List<Long> preenchimentos) |
Métodos herdados
Métodos Públicos
public static XlaSplitND <T> create ( Escopo escopo, entrada Operando <T>, N longo, Lista<Long> numSplits, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação XlaSplitND.
Parâmetros
escopo | escopo atual |
---|---|
entrada | 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 do XlaSplitND
preenchimentos XlaSplitND.Options estáticos públicos (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. |
---|