ReadVariableXlaSplitND

classe finale pubblica ReadVariableXlaSplitND

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_splits`:
[2, 2]
 
e `paddings`:
[1, 1]
 
gli `output` previsti 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>>
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
scopo 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
ritorna
  • una nuova istanza di ReadVariableXlaSplitND

iteratore pubblico< operando <T>> iteratore ()

elenco pubblico< Uscita <T>> uscite ()

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.