flux tensoriel : : opérations : : SparseAdd

#include <sparse_ops.h>

Ajoute deux objets SparseTensor pour produire un autre SparseTensor .

Résumé

Les indices des objets SparseTensor d'entrée sont supposés classés dans l'ordre lexicographique standard. Si ce n'est pas le cas, avant cette étape, exécutez SparseReorder pour restaurer l'ordre des index.

Par défaut, si la somme de deux valeurs est égale à zéro à un index, le SparseTensor de sortie inclura toujours cet emplacement particulier dans son index, stockant un zéro dans l'emplacement de valeur correspondant. Pour remplacer cela, les appelants peuvent spécifier thresh , indiquant que si la somme a une ampleur strictement inférieure à thresh , sa valeur et son index correspondants ne seront alors pas inclus. En particulier, thresh == 0 (par défaut) signifie que tout est conservé et que le seuil réel n'a lieu que pour une valeur positive.

Dans les formes suivantes, nnz est le décompte après prise en compte thresh .

Arguments :

  • scope : un objet Scope
  • a_indices : 2-D. Les indices du premier SparseTensor , taille [nnz, ndims] Matrix.
  • a_values : 1-D. Les values du premier SparseTensor , taille [nnz] Vector.
  • a_shape : 1-D. La shape du premier SparseTensor , taille [ndims] Vector.
  • b_indices : 2-D. Les indices du deuxième SparseTensor , taille [nnz, ndims] Matrix.
  • valeurs b_: 1-D. Les values du deuxième SparseTensor , taille [nnz] Vector.
  • forme_b : 1-D. La shape du deuxième SparseTensor , taille [ndims] Vector.
  • seuil : 0-D. Seuil d'amplitude qui détermine si une paire valeur de sortie/index prend de la place.

Retours :

Constructeurs et Destructeurs

SparseAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b_indices, :: tensorflow::Input b_values, :: tensorflow::Input b_shape, :: tensorflow::Input thresh)

Attributs publics

operation
sum_indices
sum_shape
sum_values

Attributs publics

opération

Operation operation

somme_indices

::tensorflow::Output sum_indices

somme_forme

::tensorflow::Output sum_shape

somme_valeurs

::tensorflow::Output sum_values

Fonctions publiques

SparseAdd

 SparseAdd(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input a_values,
  ::tensorflow::Input a_shape,
  ::tensorflow::Input b_indices,
  ::tensorflow::Input b_values,
  ::tensorflow::Input b_shape,
  ::tensorflow::Input thresh
)