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

org.tensorflow.op.RawOp
booleano finale
è uguale a (Oggetto oggetto)
finale int
codice hash ()
Operazione
operazione ()
Restituisce questa unità di calcolo come una singola Operation .
stringa finale
accordare ()
booleano
è uguale a (Oggetto arg0)
Classe finale<?>
getClass ()
int
codice hash ()
vuoto finale
notificare ()
vuoto finale
notificaTutti ()
Corda
accordare ()
vuoto finale
attendere (lungo arg0, int arg1)
vuoto finale
aspetta (lungo arg0)
vuoto finale
Aspettare ()
org.tensorflow.op.Op
ambiente di esecuzione astratto
ambiente ()
Restituisce l'ambiente di esecuzione in cui è stata creata questa operazione.
Operazione astratta
operazione ()
Restituisce questa unità di calcolo come una singola Operation .
org.tensorflow.Operand
Output astratto <T>
comeuscita ()
Restituisce l'handle simbolico del tensore.
astratto T
come tensore ()
Restituisce il tensore in questo operando.
forma astratta
forma ()
Restituisce la forma (possibilmente parzialmente conosciuta) del tensore a cui fa riferimento l' Output di questo operando.
Classe astratta<T>
tipo ()
Restituisce il tipo tensore di questo operando
org.tensorflow.ndarray.Shaped
astratto int
rango ()
forma astratta
forma ()
astratto lungo
misurare ()
Calcola e restituisce la dimensione totale di questo contenitore, in numero di valori.

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.