tensorflow :: opérations :: SparseMatMul
#include <math_ops.h>
Multipliez la matrice "a" par la matrice "b".
Résumé
Les entrées doivent être des matrices bidimensionnelles et la dimension intérieure de "a" doit correspondre à la dimension extérieure de "b". «A» et «b» doivent tous deux être des Tensor
et non des SparseTensor
. Cette opération est optimisée pour le cas où au moins l'un de "a" ou "b" est clairsemé, dans le sens où ils ont une grande proportion de valeurs nulles. Le seuil de rentabilité pour utiliser cela par rapport à une multiplication de matrice dense sur une plate-forme était de 30% de valeurs nulles dans la matrice clairsemée.
Le calcul du gradient de cette opération ne profitera de la rareté du gradient d'entrée que lorsque ce gradient provient d'un Relu .
Arguments:
- scope: un objet Scope
Retour:
-
Output
: le tenseur du produit.
Constructeurs et destructeurs | |
---|---|
SparseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b) | |
SparseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a, :: tensorflow::Input b, const SparseMatMul::Attrs & attrs) |
Attributs publics | |
---|---|
operation | |
product |
Fonctions publiques | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Fonctions statiques publiques | |
---|---|
AIsSparse (bool x) | |
BIsSparse (bool x) | |
TransposeA (bool x) | |
TransposeB (bool x) |
Structs | |
---|---|
tensorflow :: ops :: SparseMatMul :: Attrs | Paramètres d' attributs facultatifs pour SparseMatMul . |
Attributs publics
opération
Operation operation
produit
::tensorflow::Output product
Fonctions publiques
SparseMatMul
SparseMatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a, ::tensorflow::Input b )
SparseMatMul
SparseMatMul( const ::tensorflow::Scope & scope, ::tensorflow::Input a, ::tensorflow::Input b, const SparseMatMul::Attrs & attrs )
nœud
::tensorflow::Node * node() const
operator :: tensorflow :: Input
operator::tensorflow::Input() const
operator :: tensorflow :: Output
operator::tensorflow::Output() const
Fonctions statiques publiques
AIsSparse
Attrs AIsSparse( bool x )
BIsSparse
Attrs BIsSparse( bool x )
TransposerA
Attrs TransposeA( bool x )
TransposerB
Attrs TransposeB( bool x )