Suddividi

lezione finale pubblica Spalato

Divide un tensore in tensori `num_split` lungo una dimensione.

Metodi pubblici

<T> statico Diviso <T>
create ( ambito ambito , asse operando <intero>, valore operando <T>, numSplit lungo)
Metodo factory per creare una classe che racchiude una nuova operazione Split.
Iteratore< Operando <T>>
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

iteratore pubblico< operando <T>> iteratore ()

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