tensorflow:: ops:: SparseConcat
#include <sparse_ops.h> Łączy listę obiektów SparseTensor wzdłuż określonego wymiaru.
Streszczenie
Konkatenacja odbywa się w odniesieniu do gęstych wersji tych rzadkich tensorów. Zakłada się, że każde wejście jest SparseTensor , którego elementy są uporządkowane wzdłuż rosnącej liczby wymiarów.
Kształty wszystkich danych wejściowych muszą być takie same, z wyjątkiem wymiaru połączenia. Listy indices , values i shapes muszą mieć tę samą długość.
Kształt wyjściowy jest identyczny z kształtem wejściowym, z wyjątkiem wymiaru połączenia, gdzie jest sumą rozmiarów wejściowych wzdłuż tego wymiaru.
Elementy wyjściowe zostaną wykorzystane w celu zachowania kolejności sortowania wzdłuż rosnącej liczby wymiarów.
Ta operacja jest wykonywana w czasie O(M log M) , gdzie M to całkowita liczba niepustych wartości we wszystkich danych wejściowych. Wynika to z konieczności wewnętrznego sortowania w celu wydajnego łączenia w dowolnym wymiarze.
Na przykład, jeśli concat_dim = 1 i dane wejściowe są
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"
następnie wynik będzie
shape = [2, 7] [0, 2]: "a" [0, 4]: "d" [0, 5]: "e" [1, 0]: "b" [1, 1]: "c"
Graficznie jest to równoważne z wykonaniem
[ a] concat [ d e ] = [ a d e ] [b c ] [ ] [b c ]
Argumenty:
- zakres: obiekt zakresu
- Indeksy: 2-D. Indeksy każdego wejścia
SparseTensor. - wartości: 1-D. Niepuste wartości każdego
SparseTensor. - kształty: 1-D. Kształty każdego
SparseTensor. - concat_dim: Wymiar, który ma zostać połączony. Musi mieścić się w zakresie [-rank, rank), gdzie rank to liczba wymiarów w każdym wejściowym
SparseTensor.
Zwroty:
-
Outputoutput_indices: 2-D. Indeksy połączonegoSparseTensor. -
Outputoutput_values: 1-D. Niepuste wartości połączonegoSparseTensor. - Wyjście
Output: 1-D. Kształt połączonego obiektuSparseTensor.
Konstruktory i destruktory | |
|---|---|
SparseConcat (const :: tensorflow::Scope & scope, :: tensorflow::InputList indices, :: tensorflow::InputList values, :: tensorflow::InputList shapes, int64 concat_dim) |
Atrybuty publiczne | |
|---|---|
operation | |
output_indices | |
output_shape | |
output_values | |
Atrybuty publiczne
działanie
Operation operation
indeksy_wyjściowe
::tensorflow::Output output_indices
kształt_wyjściowy
::tensorflow::Output output_shape
wartości_wyjściowe
::tensorflow::Output output_values
Funkcje publiczne
SparseConcat
SparseConcat( const ::tensorflow::Scope & scope, ::tensorflow::InputList indices, ::tensorflow::InputList values, ::tensorflow::InputList shapes, int64 concat_dim )