Suddivide il tensore di input della variabile risorsa in tutte le dimensioni.
Un'operazione che divide il tensore di input della variabile risorsa 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]]
`num_split`: [2, 2]
e "imbottiture": [1, 1]
gli "output" attesi sono: [[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Classi nidificate
classe | LeggiVariableXlaSplitND.Options | Attributi facoltativi per ReadVariableXlaSplitND |
Metodi pubblici
statico <T> ReadVariableXlaSplitND <T> | create ( ambito ambito , operando <?> risorsa, classe<T> T, lungo N, elenco<lungo> numSplits, opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione ReadVariableXlaSplitND. |
Iteratore< Operando <T>> | iteratore () |
Elenco< Uscita <T>> | uscite () |
ReadVariableXlaSplitND.Options statico | imbottiture (Elenco<Long> imbottiture) |
Metodi ereditati
Metodi pubblici
public static ReadVariableXlaSplitND <T> create ( ambito ambito , operando <?> risorsa, classe<T> T, N lungo, Elenco<lungo> numSplits, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione ReadVariableXlaSplitND.
Parametri
ambito | ambito attuale |
---|---|
risorsa | Variabile risorsa del tensore di input da suddividere 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 ReadVariableXlaSplitND
imbottiture statiche pubbliche ReadVariableXlaSplitND.Options (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. |
---|