tensorflow::ops::SparseMatMul

Multipliez la matrice « a » par la matrice « b ».

Résumé

Les entrées doivent être des matrices bidimensionnelles et la dimension interne de « a » doit correspondre à la dimension externe de « b ». « a » et « b » doivent être Tensor (et non SparseTensor ). Cette opération est optimisée lorsque « a » ou « b » est creux, c'est-à-dire qu'elle contient une forte proportion de zéros. Le seuil de rentabilité de cette opération par rapport à la multiplication de matrices denses sur une plateforme donnée était de 30 % de zéros dans la matrice creuse.

Le calcul du gradient de cette opération ne tirera parti de la parcimonie du gradient d'entrée que lorsque ce gradient provient d'une fonction Relu .

Arguments :

• portée : un objet Scope

Retours :

• Output : Le tenseur produit.

attributs publics

Operation operation

::tensorflow::Output product

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AIsSparse(
  bool x
)

Attrs BIsSparse(
  bool x
)

Attrs TransposeA(
  bool x
)

Attrs TransposeB(
  bool x
)