TridiagonalMatMul

cours final public TridiagonalMatMul

Calculer le produit avec une matrice tridiagonale.

Calcule le produit de deux matrices, où la matrice de gauche est une matrice tridiagonale.

Méthodes publiques

Sortie <T>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
statique <T> TridiagonalMatMul <T>
créer ( Scope scope, Opérande <T> superdiag, Opérande <T> maindiag, Opérande <T> subdiag, Opérande <T> rhs)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TridiagonalMatMul.
Sortie <T>
sortir ()
Tenseur de forme `[..., M, N]` contenant le produit.

Méthodes héritées

Méthodes publiques

sortie publique <T> 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 TridiagonalMatMul <T> create ( Scope scope, Operand <T> superdiag, Operand <T> maindiag, Operand <T> subdiag, Operand <T> rhs)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération TridiagonalMatMul.

Paramètres
portée portée actuelle
superdiag Tenseur de forme `[..., 1, M]`, représentant les superdiagonales de matrices tri-diagonales à gauche de la multiplication. Le dernier élément est ignoré.
diagnostic principal Tenseur de forme `[..., 1, M]`, représentant les diagonales principales des matrices tri-diagonales à gauche de la multiplication.
sous-diagramme Tenseur de forme `[..., 1, M]`, représentant les sous-diagonales des matrices tri-diagonales à gauche de la multiplication. Le premier élément est ignoré.
rhs Tenseur de forme `[..., M, N]`, représentant les matrices MxN à droite de la multiplication.
Retour
  • une nouvelle instance de TridiagonalMatMul

sortie publique <T> sortie ()

Tenseur de forme `[..., M, N]` contenant le produit.