tf.raw_ops.SparseConcat

Объединяет список SparseTensor вдоль указанного измерения.

Просмотреть псевдонимы

Совместимые псевдонимы для миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.SparseConcat

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-D. Индексы каждого входного `SparseTensor`.
`values`	Список той же длины, что и `indices`, состоящий из объектов `Tensor` того же типа. 1-D. Непустые значения каждого `SparseTensor`.
`shapes`	Список той же длины, что и `indices`, состоящий из объектов `Tensor` типа `int64`. 1-D. Формы каждого `SparseTensor`.
`concat_dim`	Объект `int`. Измерение, вдоль которого выполняется конкатенация. Должно находиться в диапазоне [-rank, rank), где rank — количество измерений каждого входного `SparseTensor`.
`name`	Имя операции (необязательно).

Возвращаемые значения
Кортеж объектов `Tensor` (output_indices, output_values, output_shape).
`output_indices`	Объект `Tensor` типа `int64`.
`output_values`	Объект `Tensor`. Имеет тот же тип, что и `values`.
`output_shape`	Объект `Tensor` типа `int64`.

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