тензорный поток :: 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
)