Multipliez la matrice "a" par la matrice "b".
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 un des « a » ou « b » est clairsemé, dans le sens où ils ont une grande proportion de valeurs nulles. Le seuil de rentabilité pour l'utilisation de cette multiplication par rapport à une 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 parcimonie du gradient d'entrée que lorsque ce gradient provient d'un Relu.
Classes imbriquées
classe | SparseMatMul.Options | Attributs facultatifs pour SparseMatMul |
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
SparseMatMul.Options statique | aIsSparse (aIsSparse booléen) |
Sortie < TFloat32 > | comme Sortie () Renvoie le handle symbolique du tenseur. |
SparseMatMul.Options statique | bIsSparse (booléen bIsSparse) |
SparseMatMul statique | créer ( Portée de la portée , Opérande <? étend TNumber > a, Opérande < ? étend TNumber > b, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération SparseMatMul. |
Sortie < TFloat32 > | produit () |
SparseMatMul.Options statique | transposeA (transposeA booléenne) |
SparseMatMul.Options statique | transposeB (transposeB booléenne) |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
Sortie publique < TFloat32 > asOutput ()
Renvoie le handle symbolique du tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
public static SparseMatMul create ( Scope scope, Operand <? extends TNumber > a, Operand <? extends TNumber > b, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération SparseMatMul.
Paramètres
portée | portée actuelle |
---|---|
choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de SparseMatMul