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 `partial_pivoting`. En la GPU, se utiliza la biblioteca cuSPARSE de Nvidia: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Los backends XLA aún no admiten la rotación parcial.
Clases anidadas
clase | TridiagonalSolve.Opciones | Atributos opcionales para TridiagonalSolve |
Métodos públicos
Salida <T> | como salida () Devuelve el identificador simbólico de un tensor. |
<T> estática TridiagonalResolver <T> | crear ( 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. |
Salida <T> | salida () Tensor de forma `[..., M, K]` que contiene las soluciones |
TridiagonalSolve.Opciones estáticas | pivote parcial (booleano pivote parcial) |
TridiagonalSolve.Opciones estáticas | perturbSingular (booleano perturbSingular) |
Métodos Heredados
Métodos públicos
salida pública <T> como salida ()
Devuelve el identificador simbólico de un tensor.
Las entradas de 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 ( Scope scope, Operand <T> diagonals, Operand <T> rhs, Options... options)
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 que son la superdiagonal, las diagonales y las subdiagonales, en orden. Se ignora 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.Opciones pivote parcial (booleano pivote parcial)
Parámetros
parcial pivotante | Ya sea para aplicar el pivote parcial. El giro parcial hace que el procedimiento sea más estable, pero más lento. |
---|