Risolve sistemi di equazioni tridiagonali.
Risolve sistemi di equazioni tridiagonali. Supporta dimensioni batch e più lati destri per ciascun lato sinistro. Sulla CPU, la soluzione viene calcolata tramite eliminazione gaussiana con o senza pivoting 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 pivoting parziale non è ancora supportato dai backend XLA.
Classi nidificate
| classe | TridiagonalSolve.Options | Attributi facoltativi per TridiagonalSolve | |
Metodi pubblici
| Uscita <T> | comeuscita () Restituisce la maniglia simbolica di un tensore. |
| statico <T> TridiagonaleRisolvi <T> | create ( ambito ambito , operando <T> diagonali, operando <T> destra, Opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione TridiagonalSolve. |
| Uscita <T> | produzione () Tensore di forma `[..., M, K]` contenente le soluzioni |
| TridiagonalSolve.Options statico | partialPivoting (parzialePivoting booleano) |
Metodi ereditati
Metodi pubblici
Uscita pubblica <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, Operando <T> diagonali, Operando <T> dx, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude 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, le diagonali e le sottodiagonali, in ordine. L'ultimo elemento della superdiagonale e il primo elemento della subdiagonale vengono ignorati. |
| ds | Tensore di forma `[..., M, K]`, che rappresenta K lati destri per ciascun lato sinistro. |
| opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di TridiagonalSolve
public static TridiagonalSolve.Options partialPivoting (Boolean partialPivoting)
Parametri
| parzialePivoting | Se applicare il pivoting parziale. La rotazione parziale rende la procedura più stabile, ma più lenta. |
|---|