tensorflow::ops::SparseConcat

#include <sparse_ops.h>

Конкатенирует список SparseTensor по указанному измерению.

Краткое описание

Конкатенация выполняется относительно плотных версий этих разреженных тензоров. Предполагается, что каждый вход представляет собой SparseTensor, элементы которого упорядочены по возрастанию номера измерения.

Форматы всех входных данных должны совпадать, за исключением измерения конкатенации. indices, values и shapes списки должны иметь одинаковую длину.

Форма выходного тензора совпадает с форматами входных тензоров, за исключением измерения конкатенации, где она равна сумме размеров входных тензоров вдоль этого измерения.

Элементы вывода будут отсортированы для сохранения порядка сортировки по возрастанию номера измерения.

Эта операция выполняется за время 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
• indices: 2-мерный массив. Индексы каждого входного SparseTensor.
• values: 1-мерный массив. ненулевые значения каждого SparseTensor.
• shapes: 1-мерный массив. Форматы каждого SparseTensor.
• concat_dim: Измерение для конкатенации. Должно быть в диапазоне [-ранг, ранг), где ранг — число измерений в каждом входном SparseTensor.

Возвращаемые значения:

• Output output_indices: 2-мерный массив. Индексы конкатенированного SparseTensor.
• Output output_values: 1-мерный массив. ненулевые значения конкатенированного SparseTensor.
• Output output_shape: 1-мерный массив. Формат конкатенированного SparseTensor.

Общедоступные атрибуты

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
)