Spec-Zone.ru › TensorFlow 2.3

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

Spec-Zone.ru

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