Spec-Zone.ru › TensorFlow 2.4

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-мерный. Индексы каждого входного 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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API