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 pivote 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 | Atributos opcionales para TridiagonalSolve |
Métodos públicos
Salida <T> | asOutput () Devuelve el identificador simbólico de un tensor. |
estática <T> 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 |
TridiagonalSolve.Options estática | ParticularPivoting ( pivote parcial booleano) |
Métodos heredados
Métodos públicos
Salida pública <T> asOutput ()
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 ( Scope scope, Operand <T> diagonales, Operand <T> rhs, 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 ParticalPivoting (Boolean ParticularPivoting)
Parámetros
pivotante parcial | Ya sea para aplicar pivotamiento parcial. El pivote parcial hace que el procedimiento sea más estable, pero más lento. |
---|