Risolve sistemi di equazioni tridiagonali.
Risolve sistemi di equazioni tridiagonali. Supporta le dimensioni del lotto e più lati destri per ciascun lato sinistro. Sulla CPU, la soluzione viene calcolata tramite eliminazione gaussiana con o senza pivot parziale, a seconda dell'attributo `partial_pivoting`. Sulla GPU, viene utilizzata la libreria cuSPARSE di Nvidia: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Il pivot parziale non è ancora supportato dai backend XLA.
Classi annidate
classe | TridiagonalSolve.Options | Attributi opzionali per TridiagonalSolve |
Metodi pubblici
Uscita <T> | asOutput () Restituisce la maniglia simbolica di un tensore. |
static <T> TridiagonalSolve <T> | create ( Scope scope, Operando <T> diagonali, Operando <T> rhs, Opzioni ... opzioni) Metodo di fabbrica per creare una classe che avvolge una nuova operazione TridiagonalSolve. |
Uscita <T> | output () Tensore di forma "[..., M, K]" contenente le soluzioni |
static TridiagonalSolve.Options | partialPivoting (Boolean partialPivoting) |
Metodi ereditati
Metodi pubblici
output pubblico <T> asOutput ()
Restituisce la maniglia simbolica di un tensore.
Gli input per le operazioni TensorFlow sono output di un'altra operazione TensorFlow. Questo metodo viene utilizzato per ottenere un handle simbolico che rappresenta il calcolo dell'input.
public static TridiagonalSolve <T> create ( Scope scope, Operand <T> diagonals, Operand <T> rhs, Options ... options)
Metodo Factory per creare una classe che avvolge una nuova operazione TridiagonalSolve.
Parametri
scopo | ambito attuale |
---|---|
diagonali | Tensore di forma "[..., 3, M]" le cui 2 dimensioni più interne rappresentano le matrici tridiagonali con tre righe che sono la superdiagonale, diagonale e sottodiagonale, in ordine. L'ultimo elemento della superdiagonale e il primo elemento della sottodiagonale vengono ignorati. |
rhs | Tensore di forma "[..., M, K]", che rappresenta K lati destri per ogni lato sinistro. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di TridiagonalSolve
public static TridiagonalSolve.Options partialPivoting (Boolean partialPivoting)
Parametri
partialPivoting | Se applicare la rotazione parziale. La rotazione parziale rende la procedura più stabile, ma più lenta. |
---|