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.4/api_docs/python/tf/raw_ops/SparseConcat