TridiagonalSolve

clase final pública TridiagonalSolve

Resuelve sistemas tridiagonales de ecuaciones.

Resuelve sistemas tridiagonales de ecuaciones. Admite dimensiones de lote y múltiples lados derechos por cada lado izquierdo. En la CPU, la solución se calcula mediante eliminación gaussiana con o sin pivote parcial, dependiendo del atributo "partial_pivoting". En GPU, se utiliza la biblioteca cuSPARSE de Nvidia: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Los backends XLA aún no admiten el pivotamiento parcial.

Clases anidadas

clase Opciones de solución tridiagonal Atributos opcionales para TridiagonalSolve

Constantes

Cadena OP_NOMBRE El nombre de esta operación, como lo conoce el motor central de TensorFlow.

Métodos públicos

Salida <T>
como salida ()
Devuelve el identificador simbólico del tensor.
estático <T extiende TType > TridiagonalSolve <T>
crear ( Alcance alcance, Operando <T> diagonales, Operando <T> derechos, Opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación TridiagonalSolve.
Salida <T>
producción ()
Tensor de forma `[..., M, K]` que contiene las soluciones
Opciones de resolución tridiagonal estáticas
Pivote parcial (pivote parcial booleano)

Métodos heredados

Constantes

Cadena final estática pública OP_NAME

El nombre de esta operación, como lo conoce el motor central de TensorFlow.

Valor constante: "TridiagonalSolve"

Métodos públicos

Salida pública <T> como Salida ()

Devuelve el identificador simbólico del 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 ( 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.

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 ignoran 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.Options parcialPivoting (Pivoting parcial booleano)

Parámetros
parcialPivotante Si se debe aplicar un pivote parcial. El giro parcial hace que el procedimiento sea más estable, pero más lento.