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

Constantes

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

Métodos públicos

Salida <T>
asOutput ()
Devuelve el identificador simbólico del tensor.
estática <T se extiende Ttype > 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)

Métodos heredados

Constantes

OP_NAME pública final static String

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

Valor constante: "TridiagonalSolve"

Métodos públicos

pública de salida <T> asOutput ()

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> 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 pivotaje parcial. El pivote parcial hace que el procedimiento sea más estable, pero más lento.