tensoreflusso:: ops:: SparseConcat
#include <sparse_ops.h>
Concatena un elenco di SparseTensor
lungo la dimensione specificata.
Riepilogo
La concatenazione riguarda le versioni dense di questi tensori sparsi. Si presuppone che ogni input sia uno SparseTensor
i cui elementi sono ordinati secondo un numero di dimensione crescente.
Tutte le forme degli input devono corrispondere, ad eccezione della dimensione concat. Gli elenchi di indices
, values
e shapes
devono avere la stessa lunghezza.
La forma dell'output è identica a quella degli input, tranne che lungo la dimensione concat, dove è la somma delle dimensioni degli input lungo quella dimensione.
Gli elementi di output verranno utilizzati per preservare l'ordinamento lungo il numero di dimensione crescente.
Questa operazione viene eseguita nel tempo O(M log M)
, dove M
è il numero totale di valori non vuoti in tutti gli input. Ciò è dovuto alla necessità di un ordinamento interno per concatenare in modo efficiente attraverso una dimensione arbitraria.
Ad esempio, se concat_dim = 1
e gli input sono
sp_inputs[0]: shape = [2, 3] [0, 2]: "a" [1, 0]: "b" [1, 1]: "c" sp_inputs[1]: shape = [2, 4] [0, 1]: "d" [0, 2]: "e"
quindi l'output sarà
shape = [2, 7] [0, 2]: "a" [0, 4]: "d" [0, 5]: "e" [1, 0]: "b" [1, 1]: "c"
Graficamente questo equivale a fare
[ a] concat [ d e ] = [ a d e ] [b c ] [ ] [b c ]
Argomenti:
- scope: un oggetto Scope
- indici: 2-D. Indici di ogni ingresso
SparseTensor
. - valori: 1-D. Valori non vuoti di ogni
SparseTensor
. - forme: 1-D. Forme di ogni
SparseTensor
. - concat_dim: dimensione su cui concatenare. Deve essere compreso nell'intervallo [-rank, rango), dove rango è il numero di dimensioni in ciascun input
SparseTensor
.
Resi:
-
Output
output_indices: 2-D. Indici delloSparseTensor
concatenato. -
Output
valori_output: 1-D. Valori non vuoti delloSparseTensor
concatenato. -
Output
forma_output: 1-D. Forma delloSparseTensor
concatenato.
Costruttori e distruttori | |
---|---|
SparseConcat (const :: tensorflow::Scope & scope, :: tensorflow::InputList indices, :: tensorflow::InputList values, :: tensorflow::InputList shapes, int64 concat_dim) |
Attributi pubblici | |
---|---|
operation | |
output_indices | |
output_shape | |
output_values |
Attributi pubblici
operazione
Operation operation
indici_output
::tensorflow::Output output_indices
forma_output
::tensorflow::Output output_shape
valori_di_output
::tensorflow::Output output_values
Funzioni pubbliche
SparseConcat
SparseConcat( const ::tensorflow::Scope & scope, ::tensorflow::InputList indices, ::tensorflow::InputList values, ::tensorflow::InputList shapes, int64 concat_dim )