공개 최종 클래스 TridiagonalSolve
방정식의 삼중대각 시스템을 해결합니다.
방정식의 삼중대각 시스템을 해결합니다. 배치 차원과 각 왼쪽마다 여러 오른쪽 변을 지원합니다. CPU에서는 `partial_pivoting` 속성에 따라 부분 피벗 여부에 관계없이 가우스 제거를 통해 솔루션이 계산됩니다. GPU에서는 Nvidia의 cuSPARSE 라이브러리가 사용됩니다. https://docs.nvidia.com/cuda/cusparse/index.html#gtsv 부분 피버팅은 아직 XLA 백엔드에서 지원되지 않습니다.
중첩 클래스
| 수업 | TridiagonalSolve.Options | TridiagonalSolve 의 선택적 속성 | |
공개 방법
| 출력 <T> | 출력 () 텐서의 기호 핸들을 반환합니다. |
| 정적 <T> TridiagonalSolve <T> | |
| 출력 <T> | 출력 () 솔루션을 포함하는 형태 `[..., M, K]`의 텐서 |
| 정적 TridiagonalSolve.Options | 부분 피보팅 (부울 부분 피보팅) |
| 정적 TridiagonalSolve.Options | perturbSingular (부울 perturbSingular) |
상속된 메서드
공개 방법
공개 출력 <T> asOutput ()
텐서의 기호 핸들을 반환합니다.
TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.
공개 정적 TridiagonalSolve <T> 생성 ( 범위 범위, 피연산자 <T> 대각선, 피연산자 <T> rhs, 옵션... 옵션)
새로운 TridiagonalSolve 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
매개변수
| 범위 | 현재 범위 |
|---|---|
| 대각선 | 가장 안쪽 2차원이 상대각선, 대각선, 하대각선의 세 행을 순서대로 갖는 삼중대각선 행렬을 나타내는 '[..., 3, M]' 모양의 텐서입니다. 상부 대각선의 마지막 요소와 하부 대각선의 첫 번째 요소는 무시됩니다. |
| rhs | `[..., M, K]' 형태의 텐서. 각 왼쪽 변마다 K개의 오른쪽 변을 나타냅니다. |
| 옵션 | 선택적 속성 값을 전달합니다. |
보고
- TridiagonalSolve의 새로운 인스턴스
공개 정적 TridiagonalSolve.Options 부분 피보팅 (부울 부분 피보팅)
매개변수
| 부분 피벗 | 부분 피버팅을 적용할지 여부입니다. 부분 회전을 사용하면 절차가 더 안정적이지만 속도가 느려집니다. |
|---|