classe finale publique SparseMatrixMul
Multiplication élément par élément d'une matrice clairsemée avec un tenseur dense.
Renvoie une matrice clairsemée.
Le tenseur dense « b » peut être soit un scalaire ; sinon, « a » doit être un « SparseMatrix » de rang 3 ; dans ce cas, `b` doit avoir la forme `[batch_size, 1, 1]` et l'opération de multiplication est diffusée.
REMARQUE même si « b » est nul, la structure de parcimonie de la sortie ne change pas.
Méthodes publiques
| Sortie <Objet> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
| statique <T> SparseMatrixMul | créer ( Portée de portée , Opérande <?> a, Opérande <T> b) Méthode d'usine pour créer une classe encapsulant une nouvelle opération SparseMatrixMul. |
| Sortie <?> | sortir () Un tenseur de sortie dense. |
Méthodes héritées
Méthodes publiques
sortie publique <Objet> asOutput ()
Renvoie le handle symbolique d'un 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 SparseMatrixMul créer ( Scope scope, Operand <?> a, Operand <T> b)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération SparseMatrixMul.
Paramètres
| portée | portée actuelle |
|---|---|
| un | Un CSRSparseMatrix. |
| b | Un tenseur dense. |
Retour
- une nouvelle instance de SparseMatrixMul