SplitV

classe finale pubblica SplitV

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

Metodi pubblici

statico <T, U estende Numero> 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.
Iteratore< Operando <T>>
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

iteratore pubblico< operando <T>> iteratore ()

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