TridiagonaleMatMul

lezione finale pubblica TridiagonalMatMul

Calcolare il prodotto con matrice tridiagonale.

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

Metodi pubblici

Uscita <T>
comeuscita ()
Restituisce la maniglia simbolica di un tensore.
<T> statico TridiagonaleMatMul <T>
create ( scope scope, operando <T> superdiag, operando <T> maindiag, operando <T> subdiag, operando <T> rhs)
Metodo factory per creare una classe che racchiude una nuova operazione TridiagonalMatMul.
Uscita <T>
produzione ()
Tensore di forma `[..., M, N]` contenente il prodotto.

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 TridiagonalMatMul <T> create ( Scope scope, Operando <T> superdiag, Operando <T> maindiag, Operando <T> subdiag, Operando <T> rhs)

Metodo factory per creare una classe che racchiude una nuova operazione TridiagonalMatMul.

Parametri
scopo ambito attuale
superdiag Tensore di forma `[..., 1, M]`, che rappresenta le superdiagonali di matrici tridiagonali a sinistra della moltiplicazione. L'ultimo elemento viene ignorato.
maindiag Tensore di forma `[..., 1, M]`, che rappresenta le diagonali principali delle matrici tridiagonali a sinistra della moltiplicazione.
sottodiag Tensore di forma `[..., 1, M]`, che rappresenta le subdiagonali di matrici tridiagonali a sinistra della moltiplicazione. Il primo elemento viene ignorato.
ds Tensore di forma `[..., M, N]`, che rappresenta le matrici MxN a destra della moltiplicazione.
ritorna
  • una nuova istanza di TridiagonalMatMul

Uscita pubblica <T> uscita ()

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