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`. Измерение для объединения. Должно быть в диапазоне [-ранг, ранг), где ранг — это число измерений в каждом входном `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/versions/r2.9/api_docs/python/tf/raw_ops/SparseConcat