Résout les systèmes d’équations tridiagonaux.
Résout les systèmes d’équations tridiagonaux. Prend en charge les dimensions de lot et plusieurs côtés droits pour chaque côté gauche. Sur CPU, la solution est calculée via l'élimination gaussienne avec ou sans pivotement partiel, en fonction de l'attribut `partial_pivoting`. Sur GPU, la bibliothèque cuSPARSE de Nvidia est utilisée : https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Le pivotement partiel n'est pas encore pris en charge par les backends XLA.
Classes imbriquées
| classe | TridiagonalSolve.Options | Attributs facultatifs pour TridiagonalSolve | |
Constantes
| Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
| Sortie <T> | comme Sortie () Renvoie le handle symbolique du tenseur. |
| statique <T étend TType > TridiagonalSolve <T> | créer ( Portée , Opérande <T> diagonales, Opérande <T> droite, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération TridiagonalSolve. |
| Sortie <T> | sortir () Tenseur de forme `[..., M, K]` contenant les solutions |
| TridiagonalSolve.Options statique | partialPivoting (booléen partialPivoting) |
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 <T> 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 TridiagonalSolve <T> create ( Portée de la portée , Diagonales de l'Opérande <T>, Opérande <T> rhs, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération TridiagonalSolve.
Paramètres
| portée | portée actuelle |
|---|---|
| diagonales | Tenseur de forme `[..., 3, M]` dont les 2 dimensions les plus intérieures représentent les matrices tridiagonales avec trois lignes étant la superdiagonale, les diagonales et les sous-diagonales, dans l'ordre. Le dernier élément de la superdiagonale et le premier élément de la sous-diagonale sont ignorés. |
| rhs | Tenseur de forme `[..., M, K]`, représentant K côtés droits pour chaque côté gauche. |
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de TridiagonalSolve
public statique TridiagonalSolve.Options partialPivoting (booléen partialPivoting)
Paramètres
| Pivotement partiel | S'il faut appliquer un pivotement partiel. Le pivotement partiel rend la procédure plus stable, mais plus lente. |
|---|