TridiagonalMatMul

public final class TridiagonalMatMul

Calcola prodotto con matrice tridiagonale.

Calcola il prodotto di due matrici, dove la matrice sinistra è una matrice tridiagonale.

Metodi pubblici

Uscita <T>
asOutput ()
Restituisce la maniglia simbolica di un tensore.
statico <T> TridiagonalMatMul <T>
create ( ambito ambito, operando <T> superdiag, operando <T> maindiag, operando <T> subdiag, operando <T> rhs)
Metodo di fabbrica per creare una classe che avvolge una nuova operazione TridiagonalMatMul.
Uscita <T>
output ()
Tensore di forma "[..., M, N]" contenente il prodotto.

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 TridiagonalMatMul <T> create ( ambito ambito, operando <T> superdiag, operando <T> maindiag, operando <T> subdiag, operando <T> rhs)

Metodo di fabbrica per creare una classe che avvolge una nuova operazione TridiagonalMatMul.

Parametri
scopo ambito attuale
superdiag Tensore di forma "[..., 1, M]", che rappresenta le superdiagonali di matrici tri-diagonali a sinistra della moltiplicazione. L'ultimo elemento viene ignorato.
maindiag Tensore di forma "[..., 1, M]", che rappresenta le diagonali principali di matrici tri-diagonali a sinistra della moltiplicazione.
subdiag Tensore di forma "[..., 1, M]", che rappresenta sottodiagonali di matrici tri-diagonali a sinistra della moltiplicazione. Il primo elemento viene ignorato.
rhs Tensore di forma "[..., M, N]", che rappresenta matrici MxN a destra della moltiplicazione.
ritorna
  • una nuova istanza di TridiagonalMatMul

output pubblico <T> output ()

Tensore di forma "[..., M, N]" contenente il prodotto.