flux tensoriel : : opérations : : SparseConcat

Concatène une liste de SparseTensor le long de la dimension spécifiée.

Résumé

La concaténation concerne les versions denses de ces tenseurs clairsemés. On suppose que chaque entrée est un SparseTensor dont les éléments sont ordonnés selon un nombre de dimension croissant.

Les formes de toutes les entrées doivent correspondre, à l'exception de la dimension concat. Les listes indices , values et shapes doivent avoir la même longueur.

La forme de sortie est identique à celle des entrées, sauf le long de la dimension concaténée, où il s'agit de la somme des tailles des entrées le long de cette dimension.

Les éléments de sortie seront utilisés pour conserver l'ordre de tri en fonction du nombre de dimensions croissant.

Cette opération s'exécute en temps O(M log M) , où M est le nombre total de valeurs non vides sur toutes les entrées. Cela est dû à la nécessité d'un tri interne afin de concaténer efficacement sur une dimension arbitraire.

Par exemple, si concat_dim = 1 et que les entrées sont

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"

alors la sortie sera

shape = [2, 7]
[0, 2]: "a"
[0, 4]: "d"
[0, 5]: "e"
[1, 0]: "b"
[1, 1]: "c"

Graphiquement, cela équivaut à faire

[    a] concat [  d e  ] = [    a   d e  ]
[b c  ]        [       ]   [b c          ]

Arguments :

• scope : un objet Scope
• indices : 2-D. Indices de chaque entrée SparseTensor .
• valeurs : 1-D. Valeurs non vides de chaque SparseTensor .
• formes : 1-D. Formes de chaque SparseTensor .
• concat_dim : Dimension à concaténer. Doit être compris dans la plage [-rank, Rank), où Rank est le nombre de dimensions dans chaque SparseTensor d'entrée.

Retours :

• Output Output_indices : 2-D. Indices du SparseTensor concaténé.
• Output valeurs_de_sortie : 1-D. Valeurs non vides du SparseTensor concaténé.
• Output output_shape : 1-D. Forme du SparseTensor concaténé.

Attributs publics

Operation operation

::tensorflow::Output output_indices

::tensorflow::Output output_shape

::tensorflow::Output output_values

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