Resuelve sistemas tridiagonales de ecuaciones.
Resuelve sistemas tridiagonales de ecuaciones. Admite dimensiones de lote y múltiples lados derechos por cada lado izquierdo. En la CPU, la solución se calcula mediante eliminación gaussiana con o sin pivote parcial, dependiendo del atributo "partial_pivoting". En GPU, se utiliza la biblioteca cuSPARSE de Nvidia: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Los backends XLA aún no admiten el pivotamiento parcial.
Clases anidadas
clase | Opciones de solución tridiagonal | Atributos opcionales para TridiagonalSolve |
Métodos públicos
Salida <T> | como salida () Devuelve el identificador simbólico de un tensor. |
estática <T> TridiagonalSolución <T> | crear ( Alcance alcance, Operando <T> diagonales, Operando <T> derechos, Opciones... opciones) Método de fábrica para crear una clase que envuelve una nueva operación TridiagonalSolve. |
Salida <T> | producción () Tensor de forma `[..., M, K]` que contiene las soluciones |
Opciones de resolución tridiagonal estáticas | Pivote parcial (pivote parcial booleano) |
Opciones de resolución tridiagonal estáticas | perturbSingular (perturbSingular booleano) |
Métodos heredados
Métodos públicos
Salida pública <T> como Salida ()
Devuelve el identificador simbólico de un 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> create ( Alcance alcance, Operando <T> diagonales, Operando <T> rhs, Opciones... opciones)
Método de fábrica para crear una clase que envuelve 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 ignoran el último elemento de la superdiagonal y el primer elemento de la subdiagonal. |
derecho | 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
público estático TridiagonalSolve.Options parcialPivoting (Pivoting parcial booleano)
Parámetros
parcialPivotante | Si se debe aplicar un pivote parcial. El giro parcial hace que el procedimiento sea más estable, pero más lento. |
---|