Spec-Zone.ru › TensorFlow C++ 2.9

tensorflow::ops::SparseConcat

#include <sparse_ops.h>

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

Сводка

Объединение выполняется относительно плотных версий этих разреженных тензоров. Предполагается, что каждый вход является 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          ]

Аргументы:

  • scope: Объект Scope
  • indices: 2-мерный массив. Индексы каждого входного SparseTensor.
  • values: 1-мерный массив. ненулевые значения каждого SparseTensor.
  • shapes: 1-мерный массив. Форматы каждого SparseTensor.
  • concat_dim: Измерение для объединения. Должно находиться в диапазоне [-rank, rank), где rank — число измерений в каждом входном SparseTensor.

Возвращаемые значения:

  • Output output_indices: 2-мерный массив. Индексы объединённых SparseTensor.
  • Output output_values: 1-мерный массив. ненулевые значения объединённого SparseTensor.
  • Output output_shape: 1-мерный массив. Формат объединённого SparseTensor.
Конструкторы и деструкторы
SparseConcat(const ::tensorflow::Scope & scope, ::tensorflow::InputList indices, ::tensorflow::InputList values, ::tensorflow::InputList shapes, int64 concat_dim)
Публичные атрибуты
operation
Operation
output_indices
::tensorflow::Output
output_shape
::tensorflow::Output
output_values
::tensorflow::Output

Публичные атрибуты

операция

Operation operation

индексы_выхода

::tensorflow::Output output_indices

форма_выхода

::tensorflow::Output output_shape

значения_выхода

::tensorflow::Output output_values

Публичные функции

SparseConcat

 SparseConcat(
  const ::tensorflow::Scope & scope,
  ::tensorflow::InputList indices,
  ::tensorflow::InputList values,
  ::tensorflow::InputList shapes,
  int64 concat_dim
)

© 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/cc/class/tensorflow/ops/sparse-concat

Spec-Zone.ru

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