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.
Возвращаемые значения:
-
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 )
© 2020 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.4/api_docs/cc/class/tensorflow/ops/sparse-concat