Divide um tensor em tensores `num_split` ao longo de uma dimensão.
Métodos Públicos
estática <T, U estende Number> SplitV <T> | |
Iterator < Operando <T >> | iteração () |
List < Output <T >> | saída () Tensores cujo formato corresponde ao de `valor`, exceto ao longo do` eixo`, onde seus tamanhos são `size_splits [i]`. |
Métodos herdados
Métodos Públicos
public static SplitV <T> create ( Scope escopo, Operando <T> valor, operando <U> sizeSplits, Operando <Integer> eixo, Long numSplit)
Método de fábrica para criar uma classe envolvendo uma nova operação SplitV.
Parâmetros
alcance | escopo atual |
---|---|
valor | O tensor a ser dividido. |
sizeSplits | lista contendo os tamanhos de cada tensor de saída ao longo da dimensão de divisão. Deve somar a dimensão do valor ao longo do split_dim. Pode conter um -1 indicando que a dimensão deve ser inferida. |
eixo | 0-D. A dimensão ao longo da qual dividir. Deve estar no intervalo `[-rank (value), rank (value))`. |
Devoluções
- uma nova instância de SplitV
Lista pública < Output <T >> saída ()
Tensores cuja forma corresponde a de `valor`, exceto ao longo do` eixo`, onde seus tamanhos são `size_splits [i]`.