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 via eliminação Gaussiana com ou sem pivotamento parcial, dependendo do atributo `parcial_pivotamento`. 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 |
Constantes
Fragmento | OP_NAME | O nome desta operação, conforme conhecido pelo motor principal TensorFlow |
Métodos Públicos
Output <T> | asOutput () Retorna o identificador simbólico do tensor. |
estática <T estende TType > 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) |
Métodos herdados
Constantes
nome_op final String public static
O nome desta operação, conforme conhecido pelo motor principal TensorFlow
Métodos Públicos
pública Output <T> asOutput ()
Retorna o identificador simbólico do 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
public static TridiagonalSolve.Options partialPivoting (Boolean partialPivoting)
Parâmetros
partialPivoting | Se deve aplicar o pivotamento parcial. A rotação parcial torna o procedimento mais estável, mas mais lento. |
---|