Resuelve sistemas de ecuaciones tridiagonales.
Resuelve sistemas de ecuaciones tridiagonales. Admite dimensiones de lote y múltiples lados derechos por cada lado izquierdo. En la CPU, la solución se calcula a través de la eliminación gaussiana con o sin pivoteo parcial, según el atributo "pivote_parcial". En la GPU, se utiliza la biblioteca cuSPARSE de Nvidia: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Los backends de XLA aún no admiten la rotación parcial.
Clases anidadas
| clase | TridiagonalSolve.Options | Los atributos opcionales para TridiagonalSolve | |
Constantes
| Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
| Salida <T> | asOutput () Devuelve el identificador simbólico del tensor. |
| estática <T se extiende Ttype > TridiagonalSolve <T> | crear ( Alcance alcance, operando <T> diagonales, operando <T> lado derecho, Opciones ... Opciones) Método de fábrica para crear una clase que envuelva una nueva operación TridiagonalSolve. |
| Salida <T> | salida () Tensor de forma "[..., M, K]" que contiene las soluciones |
| estáticas TridiagonalSolve.Options | partialPivoting (Boolean partialPivoting) |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Métodos públicos
pública de salida <T> asOutput ()
Devuelve el identificador simbólico del tensor.
Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
public static TridiagonalSolve <T> crear ( Alcance alcance, operando <T> diagonales, operando <T> lado derecho, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación TridiagonalSolve.
Parámetros
| alcance | alcance actual |
|---|---|
| diagonales | Tensor de forma '[..., 3, M]' cuyas 2 dimensiones más internas representan las matrices tridiagonales con tres filas siendo las superdiagonales, diagonales y subdiagonales, en orden. Se ignora el último elemento de la superdiagonal y el primer elemento de la subdiagonal. |
| rhs | Tensor de forma `[..., M, K]`, que representa K lados derechos por cada lado izquierdo. |
| opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de TridiagonalSolve
public static TridiagonalSolve.Options partialPivoting (booleano partialPivoting)
Parámetros
| pivotante parcial | Ya sea para aplicar pivotaje parcial. El pivote parcial hace que el procedimiento sea más estable, pero más lento. |
|---|