Advertencia: esta API está obsoleta y se eliminará en una versión futura de TensorFlow una vez que el reemplazo sea ​​estable.

TridiagonalSolve

TridiagonalSolve public final class

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 Los 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
estáticas TridiagonalSolve.Options
partialPivoting (Boolean partialPivoting)
estáticas TridiagonalSolve.Options
perturbSingular (Boolean perturbSingular)

Métodos heredados

Métodos públicos

pública de salida <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> 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.

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

pública de salida <T> salida ()

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

public static TridiagonalSolve.Options partialPivoting (booleano partialPivoting)

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.

public static TridiagonalSolve.Options perturbSingular (booleano perturbSingular)