тензорный поток :: ops :: SparseConcat
#include <sparse_ops.h>
SparseTensor
список SparseTensor
по указанному измерению.
Резюме
Конкатенация относится к плотным версиям этих разреженных тензоров. Предполагается, что каждый вход является SparseTensor
, элементы которого упорядочены по возрастающему размерному числу.
Все формы входов должны совпадать, за исключением измерения concat. Списки indices
, values
и shapes
должны иметь одинаковую длину.
Форма вывода идентична входной, за исключением измерения concat, где это сумма размеров входов по этому измерению.
Выходные элементы будут пересортированы, чтобы сохранить порядок сортировки по возрастанию размерного числа.
Эта операция выполняется за время O(M log M)
, где M
- общее количество непустых значений на всех входах. Это связано с необходимостью внутренней сортировки для эффективной конкатенации по произвольному измерению.
Например, если concat_dim = 1
и входы
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"
тогда вывод будет
shape = [2, 7] [0, 2]: "a" [0, 4]: "d" [0, 5]: "e" [1, 0]: "b" [1, 1]: "c"
Графически это эквивалентно выполнению
[ a] concat [ d e ] = [ a d e ] [b c ] [ ] [b c ]
Аргументы:
- scope: Объект Scope
- индексы: 2-D. Индексы каждого входа
SparseTensor
. - значения: 1-D.
SparseTensor
значения каждогоSparseTensor
. - формы: 1-D. Формы каждого
SparseTensor
. - concat_dim: Измерение для объединения. Должен быть в диапазоне [-rank, rank), где rank - это количество измерений в каждом входном
SparseTensor
.
Возврат:
-
Output
output_indices: 2-D. Индексы объединенногоSparseTensor
. -
Output
output_values: 1-D.SparseTensor
значения объединенногоSparseTensor
. -
Output
output_shape: 1-D. Форма объединенногоSparseTensor
.
Конструкторы и деструкторы | |
---|---|
SparseConcat (const :: tensorflow::Scope & scope, :: tensorflow::InputList indices, :: tensorflow::InputList values, :: tensorflow::InputList shapes, int64 concat_dim) |
Публичные атрибуты | |
---|---|
operation | |
output_indices | |
output_shape | |
output_values |
Публичные атрибуты
операция
Operation operation
output_indices
::tensorflow::Output output_indices
output_shape
::tensorflow::Output output_shape
output_values
::tensorflow::Output output_values
Публичные функции
SparseConcat
SparseConcat( const ::tensorflow::Scope & scope, ::tensorflow::InputList indices, ::tensorflow::InputList values, ::tensorflow::InputList shapes, int64 concat_dim )