TridiagonalSolve

TridiagonalSolve classe final pública

Resolve sistemas tridiagonais de equações.

Resolve sistemas tridiagonais de equações. Suporta dimensões de lote e vários lados direitos para cada lado esquerdo. Na CPU, a solução é calculada através da eliminação Gaussiana com ou sem pivotamento parcial, dependendo do atributo `partial_pivoting`. Na GPU, a biblioteca cuSPARSE da Nvidia é usada: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv O pivotamento parcial ainda não é compatível com back-ends XLA.

Classes aninhadas

classe TridiagonalSolve.Options Atributos opcionais para TridiagonalSolve

Métodos Públicos

Output <T>
asOutput ()
Retorna o identificador simbólico de um tensor.
estática <T> TridiagonalSolve <T>
criar ( Scope escopo, Operando <T> diagonais, Operando <T> rhs, Options ... Opções)
Método de fábrica para criar uma classe envolvendo uma nova operação TridiagonalSolve.
Output <T>
saída ()
Tensor de forma `[..., M, K]` contendo as soluções
estáticos TridiagonalSolve.Options
partialPivoting (booleano partialPivoting)
estáticos TridiagonalSolve.Options
perturbSingular (booleano perturbSingular)

Métodos herdados

Métodos Públicos

pública Output <T> asOutput ()

Retorna o identificador simbólico de um tensor.

As entradas para as operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.

public static TridiagonalSolve <T> create ( Scope escopo, Operando <T> diagonais, Operando <T> rhs, Options ... Opções)

Método de fábrica para criar uma classe envolvendo uma nova operação TridiagonalSolve.

Parâmetros
alcance escopo atual
diagonais Tensor de forma `[..., 3, M]` cujas 2 dimensões mais internas representam as matrizes tridiagonais com três linhas sendo as superdiagonais, diagonais e subdiagonais, em ordem. O último elemento do superdiagonal e o primeiro elemento do subdiagonal são ignorados.
rhs Tensor de forma `[..., M, K]`, representando K lados direitos para cada lado esquerdo.
opções carrega valores de atributos opcionais
Devoluções
  • uma nova instância de TridiagonalSolve

pública Output <T> de saída ()

Tensor de forma `[..., M, K]` contendo as soluções

public static TridiagonalSolve.Options partialPivoting (Boolean partialPivoting)

Parâmetros
partialPivoting Se deve aplicar pivotamento parcial. A rotação parcial torna o procedimento mais estável, mas mais lento.

public static TridiagonalSolve.Options perturbSingular (Boolean perturbSingular)