тензорный поток:: опс:: РазреженныйКонкат
#include <sparse_ops.h>
Объединяет список 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 ]
Аргументы:
- область: объект области.
- индексы: 2-D. Индексы каждого входа
SparseTensor
. - значения: 1-D. Непустые значения каждого
SparseTensor
. - формы: 1-D. Формы каждого
SparseTensor
. - concat_dim: измерение для объединения. Должно находиться в диапазоне [-rank, Rank), где Rank — это количество измерений в каждом входном
SparseTensor
.
Возврат:
-
Output
выходные_индексы: 2-D. Индексы объединенногоSparseTensor
. -
Output
выходные_значения: 1-D. Непустые значения объединенногоSparseTensor
. -
Output
форма_вывода: 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
выходные_индексы
::tensorflow::Output output_indices
выходная_форма
::tensorflow::Output output_shape
выходные_значения
::tensorflow::Output output_values
Общественные функции
РазреженныйКонкат
SparseConcat( const ::tensorflow::Scope & scope, ::tensorflow::InputList indices, ::tensorflow::InputList values, ::tensorflow::InputList shapes, int64 concat_dim )