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:

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
)