Suddivide il tensore di input in tutte le dimensioni.
Un'operazione che suddivide il tensore di input in base all'attributo num_splits specificato, riempie le fette facoltativamente e restituisce le fette. Le sezioni vengono restituite in ordine di riga principale.
Questa operazione può essere generata tramite il bridge TPU.
Ad esempio, con il tensore "input":
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
[2, 2]
[1, 1]
[[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Classi nidificate
classe | XlaSplitND.Opzioni | Attributi facoltativi per XlaSplitND |
Metodi pubblici
statico <T> XlaSplitND <T> | create ( ambito ambito , operando <T> input, N lungo, List<Long> numSplits, Opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione XlaSplitND. |
Iteratore< Operando <T>> | iteratore () |
Elenco< Uscita <T>> | uscite () |
XlaSplitND.Options statico | imbottiture (Elenco<Long> imbottiture) |
Metodi ereditati
Metodi pubblici
public static XlaSplitND <T> create ( ambito ambito , operando <T> input, N lungo, Elenco<Long> numSplits, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione XlaSplitND.
Parametri
portata | ambito attuale |
---|---|
ingresso | Tensore di input da dividere in tutte le dimensioni. } out_arg { nome: "uscite" descrizione: < |
numSplits | Numero di modi per dividere per dimensione. Le dimensioni della forma devono essere equamente divisibili. |
opzioni | trasporta valori di attributi opzionali |
Ritorni
- una nuova istanza di XlaSplitND
imbottiture XlaSplitND.Options statiche pubbliche (imbottiture List<Long>)
Parametri
imbottiture | Elenco facoltativo di riempimenti giusti per dimensione del tensore di input da applicare prima della divisione. Questo può essere utilizzato per rendere una dimensione equamente divisibile. |
---|