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 premierSparseTensor
, taille[nnz, ndims]
Matrix. - a_values : 1-D. Les
values
du premierSparseTensor
, taille[nnz]
Vector. - a_shape : 1-D. La
shape
du premierSparseTensor
, taille[ndims]
Vector. - b_indices : 2-D. Les
indices
du deuxièmeSparseTensor
, taille[nnz, ndims]
Matrix. - valeurs b_: 1-D. Les
values
du deuxièmeSparseTensor
, taille[nnz]
Vector. - forme_b : 1-D. La
shape
du deuxièmeSparseTensor
, taille[ndims]
Vector. - seuil : 0-D. Seuil d'ampleur 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 )