classe finale pubblica XlaSplitND
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
| scopo | 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 |
ritorna
- 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. |
|---|