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 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

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

salida pública <T> salida ()

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

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.