TridiagonalSolve

공개 최종 클래스 TridiagonalSolve

방정식의 삼중대각 시스템을 해결합니다.

방정식의 삼중대각 시스템을 해결합니다. 배치 차원과 각 왼쪽마다 여러 오른쪽 변을 지원합니다. CPU에서는 `partial_pivoting` 속성에 따라 부분 피벗 여부에 관계없이 가우스 제거를 통해 솔루션이 계산됩니다. GPU에서는 Nvidia의 cuSPARSE 라이브러리가 사용됩니다: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv 부분 피버팅은 아직 XLA 백엔드에서 지원되지 않습니다.

중첩 클래스

수업 TridiagonalSolve.Options TridiagonalSolve 의 선택적 속성

상수

OP_NAME TensorFlow 코어 엔진에서 알려진 이 작업의 이름

공개 방법

출력 <T>
출력 ()
텐서의 기호 핸들을 반환합니다.
static <T는 TType을 확장합니다. > TridiagonalSolve <T>
생성 ( 범위 범위, 피연산자 <T> 대각선, 피연산자 <T> rhs, 옵션... 옵션)
새로운 TridiagonalSolve 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.
출력 <T>
출력 ()
솔루션을 포함하는 형태 `[..., M, K]`의 텐서
정적 TridiagonalSolve.Options
부분 피보팅 (부울 부분 피보팅)

상속된 메서드

상수

공개 정적 최종 문자열 OP_NAME

TensorFlow 코어 엔진에서 알려진 이 작업의 이름

상수 값: "TridiagonalSolve"

공개 방법

공개 출력 <T> asOutput ()

텐서의 기호 핸들을 반환합니다.

TensorFlow 작업에 대한 입력은 다른 TensorFlow 작업의 출력입니다. 이 메서드는 입력 계산을 나타내는 기호 핸들을 얻는 데 사용됩니다.

공개 정적 TridiagonalSolve <T> 생성 ( 범위 범위, 피연산자 <T> 대각선, 피연산자 <T> rhs, 옵션... 옵션)

새로운 TridiagonalSolve 작업을 래핑하는 클래스를 생성하는 팩토리 메서드입니다.

매개변수
범위 현재 범위
대각선 가장 안쪽 2차원이 상대각선, 대각선, 하대각선의 세 행을 순서대로 갖는 삼중대각선 행렬을 나타내는 '[..., 3, M]' 모양의 텐서입니다. 상부 대각선의 마지막 요소와 하부 대각선의 첫 번째 요소는 무시됩니다.
rhs `[..., M, K]' 형태의 텐서. 각 왼쪽 변마다 K개의 오른쪽 변을 나타냅니다.
옵션 선택적 속성 값을 전달합니다.
보고
  • TridiagonalSolve의 새로운 인스턴스

공개 출력 <T> 출력 ()

솔루션을 포함하는 형태 `[..., M, K]`의 텐서

공개 정적 TridiagonalSolve.Options 부분 피보팅 (부울 부분 피보팅)

매개변수
부분 피벗 부분 피버팅을 적용할지 여부입니다. 부분 회전을 사용하면 절차가 더 안정적이지만 속도가 느려집니다.