tensorflow::ops::SparseConcat

#include <sparse_ops.h>

Concaténe une liste de SparseTensor selon la dimension spécifiée.

Résumé

La concaténation s'effectue sur les versions denses de ces tenseurs creux. On suppose que chaque entrée est un SparseTensor dont les éléments sont ordonnés par ordre croissant de dimension.

Tous les formats des entrées doivent correspondre, à l'exception de la dimension de concaténation. Les listes d' indices , values et shapes doivent avoir la même longueur.

La forme de la sortie est identique à celle des entrées, sauf le long de la dimension de concaténation, où elle correspond à la somme des tailles des entrées le long de cette dimension.

Les éléments de sortie seront réorganisés afin de préserver l'ordre de tri en fonction du numéro de dimension croissant.

Cette opération s'exécute en O(M log M) , où M représente le nombre total de valeurs non vides parmi toutes les entrées. Ceci est dû à la nécessité d'un tri interne pour une concaténation efficace 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"

le résultat sera alors

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 :

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

Retours :

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

Constructeurs et destructeurs

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

attributs publics

operation
output_indices
output_shape
output_values

attributs publics

opération 

Operation operation

indices_de_sortie 

::tensorflow::Output output_indices

forme_de_sortie 

::tensorflow::Output output_shape

valeurs_de_sortie 

::tensorflow::Output output_values

fonctions publiques

SparseConcat 

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