TridiagonalSolve

classe finale pubblica TridiagonalSolve

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

Costanti

Corda OP_NAME Il nome di questa operazione, come noto al motore principale di TensorFlow

Metodi pubblici

Uscita <T>
comeuscita ()
Restituisce l'handle simbolico del tensore.
statico <T estende TType > TridiagonalSolve <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

Costanti

Stringa finale statica pubblica OP_NAME

Il nome di questa operazione, come noto al motore principale di TensorFlow

Valore costante: "TridiagonalSolve"

Metodi pubblici

Uscita pubblica <T> asOutput ()

Restituisce l'handle simbolico del 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

Uscita pubblica <T> uscita ()

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

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.