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> | |
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