TridiagonalSolve

classe finale pubblica TridiagonalSolve

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

output pubblico <T> output ()

Tensore di forma "[..., M, K]" contenente le soluzioni

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.