Diviso

classe finale pubblica Spalato

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

Metodi pubblici

static <T> Split <T>
create ( ambito ambito, asse operando <Integer>, valore operando <T>, long numSplit)
Metodo Factory per creare una classe che avvolge una nuova operazione di divisione.
Iteratore < Operando <T>>
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

iteratore pubblico < Operando <T>> iteratore ()

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.