classe finale pubblica SplitV
Divide un tensore in tensori `num_split` lungo una dimensione.
Metodi pubblici
statico <T, U estende Numero> SplitV <T> | |
Iteratore< Operando <T>> | iteratore () |
Elenco< Uscita <T>> | produzione () Tensori la cui forma corrisponde a quella di "value" tranne che lungo "axis", dove le loro dimensioni sono "size_splits[i]". |
Metodi ereditati
Metodi pubblici
public static SplitV <T> create ( ambito ambito , operando <T> valore, operando <U> sizeSplits, operando <Integer> asse, Long numSplit)
Metodo factory per creare una classe che racchiude una nuova operazione SplitV.
Parametri
scopo | ambito attuale |
---|---|
valore | Il tensore da dividere. |
dimensioneDividi | elenco contenente le dimensioni di ciascun tensore di uscita lungo la dimensione divisa. La somma deve corrispondere alla dimensione del valore lungo split_dim. Può contenere un -1 che indica che la dimensione deve essere dedotta. |
asse | 0-D. La dimensione lungo la quale dividere. Deve essere compreso nell'intervallo "[-rank(valore), ranking(valore))". |
ritorna
- una nuova istanza di SplitV
elenco pubblico< Uscita <T>> uscita ()
Tensori la cui forma corrisponde a quella di "value" tranne che lungo "axis", dove le loro dimensioni sono "size_splits[i]".