tensorflow :: ops :: SparseSplit
#include <sparse_ops.h>
Podziel SparseTensor
na tensory num_split
wzdłuż jednego wymiaru.
Podsumowanie
Jeśli shape[split_dim]
nie jest całkowitą wielokrotnością liczby num_split
. Plasterki [0 : shape[split_dim] % num_split]
jeden dodatkowy wymiar. Na przykład, jeśli split_dim = 1
i num_split = 2
a wejście to
input_tensor = shape = [2, 7] [ a d e ] [b c ]
Graficznie tensory wyjściowe to:
output_tensor[0] = shape = [2, 4] [ a ] [b c ] output_tensor[1] = shape = [2, 3] [ d e ] [ ]
Argumenty:
- zakres: obiekt Scope
- split_dim: 0-D. Wymiar, wzdłuż którego ma nastąpić podział. Musi mieścić się w zakresie
[0, rank(shape))
. - indeksy: tensor 2-D reprezentuje indeksy tensora rzadkiego.
- wartości: tensor 1-D reprezentuje wartości tensora rzadkiego.
- kształt: 1-D. tensor reprezentuje kształt rzadkiego tensora. wskaźniki wyjściowe: lista tensorów 1-D reprezentuje indeksy wyjściowych tensorów rzadkich.
- num_split: liczba sposobów podziału.
Zwroty:
-
OutputList
output_indices -
OutputList
output_values: lista tensorów 1-D reprezentuje wartości wyjściowych tensorów rzadkich. -
OutputList
output_shape: lista tensorów 1-D reprezentuje kształt wyjściowych tensorów rzadkich.
Konstruktorzy i niszczyciele | |
---|---|
SparseSplit (const :: tensorflow::Scope & scope, :: tensorflow::Input split_dim, :: tensorflow::Input indices, :: tensorflow::Input values, :: tensorflow::Input shape, int64 num_split) |
Atrybuty publiczne | |
---|---|
operation | |
output_indices | |
output_shape | |
output_values |
Atrybuty publiczne
operacja
Operation operation
output_indices
::tensorflow::OutputList output_indices
kształt_wyjścia
::tensorflow::OutputList output_shape
wartości_wyjściowe
::tensorflow::OutputList output_values
Funkcje publiczne
SparseSplit
SparseSplit( const ::tensorflow::Scope & scope, ::tensorflow::Input split_dim, ::tensorflow::Input indices, ::tensorflow::Input values, ::tensorflow::Input shape, int64 num_split )