tensorflow::ops::SparseSplit
#include <sparse_ops.h> Suddivide un SparseTensor in num_split tensori lungo una dimensione.
Riepilogo
Se shape[split_dim] non è un multiplo intero di num_split . Slices [0 : shape[split_dim] % num_split] ottiene una dimensione extra. Ad esempio, se split_dim = 1 e num_split = 2 e l'input è
input_tensor = shape = [2, 7] [ a d e ] [b c ]
Graficamente i tensori di output sono:
output_tensor[0] = shape = [2, 4] [ a ] [b c ] output_tensor[1] = shape = [2, 3] [ d e ] [ ]
Argomentazioni:
- ambito: un oggetto Scope
- split_dim: 0-D. La dimensione lungo la quale dividere. Deve essere nell'intervallo
[0, rank(shape)). - indici: il tensore 2D rappresenta gli indici del tensore sparso.
- valori: il tensore 1-D rappresenta i valori del tensore sparso.
- forma: tensore 1-D. Rappresenta la forma del tensore sparso. indici di output: Un elenco di tensori 1-D rappresenta gli indici dei tensori sparsi di output.
- num_split: Il numero di modi per suddividere.
Resi:
-
OutputListindici_output -
OutputListoutput_values: Un elenco di tensori 1D che rappresenta i valori dei tensori sparsi di output. -
OutputListoutput_shape: Un elenco di tensori 1D che rappresenta la forma dei tensori sparsi di output.
Costruttori e distruttori | |
|---|---|
SparseSplit (const :: tensorflow::Scope & scope, :: tensorflow::Input split_dim, :: tensorflow::Input indices, :: tensorflow::Input values, :: tensorflow::Input shape, int64 num_split) |
attributi pubblici | |
|---|---|
operation | |
output_indices | |
output_shape | |
output_values | |
attributi pubblici
operazione
Operation operation
indici_di_output
::tensorflow::OutputList output_indices
forma_output
::tensorflow::OutputList output_shape
valori_di_output
::tensorflow::OutputList output_values
Funzioni pubbliche
SparseSplit
SparseSplit( const ::tensorflow::Scope & scope, ::tensorflow::Input split_dim, ::tensorflow::Input indices, ::tensorflow::Input values, ::tensorflow::Input shape, int64 num_split )