XlaConcatND

lezione finale pubblica XlaConcatND

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 restituisce il tensore unito senza riempimenti.

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]]
 
`num_splits`:
[2, 2]
 
e `paddings`:
[1, 1]
 
gli `output` previsti sono:
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]
 

Classi nidificate

classe XlaConcatND.Opzioni Attributi facoltativi per XlaConcatND

Metodi pubblici

Uscita <T>
comeuscita ()
Restituisce la maniglia simbolica di un tensore.
statico <T> XlaConcatND <T>
create ( ambito ambito , Iterable< operando <T>> input, List<Long> numConcats, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione XlaConcatND.
Uscita <T>
XlaConcatND.Options statico
imbottiture (Elenco<Long> imbottiture)

Metodi ereditati

Metodi pubblici

Uscita pubblica <T> asOutput ()

Restituisce la maniglia simbolica di un tensore.

Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.

public static XlaConcatND <T> create ( scope scope, Iterable< Operand <T>> input, List<Long> numConcats, Opzioni... opzioni)

Metodo factory per creare una classe che racchiude una nuova operazione XlaConcatND.

Parametri
scopo ambito attuale
input Inserisci le sezioni del tensore in ordine di riga principale per unirle in tutte le dimensioni. Tutti gli input devono avere la stessa forma. } out_arg { nome: "output" descrizione: <
numConcats Numero di modi per unire per dimensione.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di XlaConcatND

Uscita pubblica <T> uscita ()

imbottiture XlaConcatND.Options statiche pubbliche (imbottiture 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.