classe finale pubblica Spalato
Divide un tensore in tensori `num_split` lungo una dimensione.
Metodi pubblici
static <T> Split <T> | |
Iteratore < Operando <T>> | iteratore () |
Elenco < Uscita <T>> | output () Sono tensori di forma identica, la cui forma corrisponde a quella di "valore" tranne che lungo "asse", dove le loro dimensioni sono "values.shape [split_dim] / num_split". |
Metodi ereditati
Metodi pubblici
public static Split <T> create ( Scope scope, Operand <Integer> axis, Operand <T> value, Long numSplit)
Metodo Factory per creare una classe che avvolge una nuova operazione di divisione.
Parametri
scopo | ambito attuale |
---|---|
asse | 0-D. La dimensione lungo la quale dividersi. Deve essere compreso nell'intervallo "[-rank (value), rank (value))". |
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
public List < Output <T>> output ()
Sono tensori di forma identica, la cui forma corrisponde a quella di "valore" tranne che lungo "asse", dove le loro dimensioni sono "valori.