TridiagonalSolve

clase final pública TridiagonalSolve

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

salida pública <T> salida ()

Tensor de forma `[..., M, K]` que contiene las soluciones

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.

public static TridiagonalSolve.Options perturbSingular (Boolean perturbSingular)