tf.raw_ops.SparseConcat
Конкатенирует список SparseTensor вдоль указанного измерения.
tf.raw_ops.SparseConcat(
indices, values, shapes, concat_dim, name=None
)
Конкатенация выполняется относительно плотных версий этих разреженных тензоров. Предполагается, что каждый вход — это 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 ]
| Аргументы | |
|---|---|
indices | Список из как минимум 2 Tensor объектов с типом int64. 2-мерный. Индексы каждого входного SparseTensor. |
values | Список той же длины, что и indices, Tensor объектов с тем же типом. 1-мерный. Непустые значения каждого SparseTensor. |
shapes | Список той же длины, что и indices, Tensor объектов с типом int64. 1-мерный. Формы каждого SparseTensor. |
concat_dim | int. Измерение для конкатенации. Должно быть в диапазоне [-ранг, ранг), где ранг — число измерений в каждом входном SparseTensor. |
name | Название операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Кортеж Tensor объектов (output_indices, output_values, output_shape). | |
output_indices | Tensor типа int64. |
output_values | Tensor. Имеет тот же тип, что и values. |
output_shape | Tensor типа int64. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/SparseConcat