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: Измерение для объединения. Должно находиться в диапазоне [-rank, rank), где rank — число измерений в каждом входном
SparseTensor.
Возвращаемые значения:
-
Outputoutput_indices: 2-мерный массив. Индексы объединённыхSparseTensor. -
Outputoutput_values: 1-мерный массив. ненулевые значения объединённогоSparseTensor. -
Outputoutput_shape: 1-мерный массив. Формат объединённого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
SparseConcat( const ::tensorflow::Scope & scope, ::tensorflow::InputList indices, ::tensorflow::InputList values, ::tensorflow::InputList shapes, int64 concat_dim )
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/cc/class/tensorflow/ops/sparse-concat