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

Возвращает:

• Output output_indices: двумерный массив. Индексы объединённого SparseTensor.
• Output output_values: одномерный массив. Непустые значения объединённого SparseTensor.
• Output output_shape: одномерный массив. Форма объединённого 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
)