tensorflow::ops::SparseConcat
#include <sparse_ops.h> Concatena un elenco di SparseTensor lungo la dimensione specificata.
Riepilogo
La concatenazione avviene rispetto alle versioni dense di questi tensori sparsi. Si assume che ciascun input sia un SparseTensor i cui elementi siano ordinati in base al numero di dimensione crescente.
Tutti gli input devono avere la stessa forma, ad eccezione della dimensione di concatenazione. Gli elenchi di indices , values e shapes devono avere la stessa lunghezza.
La forma dell'output è identica a quella degli input, tranne lungo la dimensione di concatenazione, dove è la somma delle dimensioni degli input lungo tale dimensione.
Gli elementi di output verranno riorganizzati in modo da preservare l'ordine di ordinamento lungo il numero di dimensioni crescente.
Questa operazione ha una complessità temporale di 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 su 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 ]
Argomentazioni:
- ambito: un oggetto Scope
- indici: 2-D. Indici di ciascun
SparseTensordi input. - valori: 1-D. Valori non vuoti di ciascun
SparseTensor. - forme: 1-D. Forme di ciascun
SparseTensor. - concat_dim: Dimensione lungo la quale concatenare. Deve essere compresa nell'intervallo [-rank, rank), dove rank è il numero di dimensioni in ciascun
SparseTensordi input.
Resi:
-
Outputoutput_indices: Indici 2D delSparseTensorconcatenato. -
Outputoutput_values: 1-D. Valori non vuoti delSparseTensorconcatenato. -
Outputoutput_shape: 1-D. Forma delSparseTensorconcatenato.
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_di_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 )