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 `partial_pivoting`. Na GPU, a biblioteca cuSPARSE da Nvidia é usada: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv A rotação parcial ainda não é suportada por back-ends XLA.
Classes aninhadas
| aula | TridiagonalSolve.Options | Atributos opcionais para TridiagonalSolve | |
Constantes
| Corda | OP_NAME | O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow |
Métodos Públicos
| Saída <T> | asOutput () Retorna o identificador simbólico do tensor. |
| estático <T estende TType > TridiagonalSolve <T> | |
| Saída <T> | saída () Tensor de forma `[..., M, K]` contendo as soluções |
| TridiagonalSolve.Options estático | parcialPivoting (Pivot parcial booleano) |
Métodos herdados
Constantes
String final estática pública OP_NAME
O nome desta operação, conforme conhecido pelo mecanismo principal do TensorFlow
Métodos Públicos
Saída pública <T> asOutput ()
Retorna o identificador simbólico do tensor.
As entradas para 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 ( Escopo , Operando <T> diagonais, Operando <T> rhs, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação TridiagonalSolve.
Parâmetros
| escopo | escopo atual |
|---|---|
| diagonais | Tensor de forma `[..., 3, M]` cujas 2 dimensões mais internas representam as matrizes tridiagonais com três linhas sendo a superdiagonal, diagonais e subdiagonais, em ordem. O último elemento da superdiagonal e o primeiro elemento da subdiagonal são ignorados. |
| rh | 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 do TridiagonalSolve
public static TridiagonalSolve.Options parcialPivoting (Boolean parcialPivoting)
Parâmetros
| parcialPivotante | Se deve aplicar pivotamento parcial. A rotação parcial torna o procedimento mais estável, porém mais lento. |
|---|