Divide un tensore in tensori `num_split` lungo una dimensione.
Metodi pubblici
<T> statico Diviso <T> | |
Iteratore< Operando <T>> | iteratore () |
Elenco< Uscita <T>> | produzione () Sono tensori di forma identica, la cui forma corrisponde a quella di "valore" tranne che lungo "asse", dove le loro dimensioni sono "valori.forma[split_dim] / num_split". |
Metodi ereditati
Metodi pubblici
creazione di suddivisione statica pubblica <T> (ambito ambito , asse operando <Integer>, valore operando <T>, numSplit lungo)
Metodo factory per creare una classe che racchiude una nuova operazione Split.
Parametri
scopo | ambito attuale |
---|---|
asse | 0-D. La dimensione lungo la quale dividere. Deve essere compreso nell'intervallo "[-rank(valore), ranking(valore))". |
valore | Il tensore da dividere. |
numSplit | Il numero di modi per dividere. Deve dividere equamente "value.shape[split_dim]". |
ritorna
- una nuova istanza di Spalato
elenco pubblico< Uscita <T>> uscita ()
Sono tensori di forma identica, la cui forma corrisponde a quella di "valore" tranne che lungo "asse", dove le loro dimensioni sono "valori.forma[split_dim] / num_split".