classe finale pubblica AssignVariableXlaConcatND
Concatena il tensore di input su tutte le dimensioni.
Un'operazione che unisce suddivide il tensore di input in base all'attributo num_splits specificato, elimina i riempimenti facoltativamente e scrive il tensore unito senza riempimenti nella variabile di risorsa.
Questa operazione può essere generata tramite il bridge TPU.
Ad esempio, con il tensore `input`:
[[0, 1],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
[2, 2]
[1, 1]
[[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
Classi nidificate
| classe | AssegnaVariableXlaConcatND.Options | Attributi facoltativi per AssignVariableXlaConcatND | |
Metodi pubblici
| <T> statico AssignVariableXlaConcatND | create ( ambito ambito , risorsa operando <?>, ingressi Iterable< operando <T>>, elenco<lungo> numConcats, opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione AssignVariableXlaConcatND. |
| static AssignVariableXlaConcatND.Options | imbottiture (Elenco<Long> imbottiture) |
Metodi ereditati
Metodi pubblici
public static AssignVariableXlaConcatND create ( ambito ambito , operando <?> risorsa, Iterable< operando <T>> input, List<Long> numConcats, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione AssignVariableXlaConcatND.
Parametri
| scopo | ambito attuale |
|---|---|
| risorsa | Variabile di risorsa per tensori di input concatenati in tutte le dimensioni. } in_arg { nome: "ingressi" descrizione: < |
| numConcats | Numero di modi per unire per dimensione. |
| opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di AssignVariableXlaConcatND
riempimenti pubblici statici AssignVariableXlaConcatND.Options (riempimenti List<Long>)
Parametri
| imbottiture | Elenco facoltativo delle imbottiture giuste per dimensione da rimuovere dal tensore unito finale. Queste imbottiture non devono superare la dimensione della dimensione del risultato unito prima dell'eliminazione delle imbottiture. |
|---|