본문으로 건너뛰기

torch-fx-tracing

torch.fx 기호적 추적

PyTorch 모델의 실행 그래프를 기록해 모듈별 연산 순서를 재구성하는 기능이다. 선언 순서가 아닌 실제 연산 의존성에 따른 실행 순서를 얻음으로써 모듈의 계산 깊이(depth)를 정확히 파악하고, 이 정보를 기준으로 모듈별 통계와 그래디언트 플롯의 x축을 정렬한다. 따라서 재귀적·조건부 실행이 있는 모델에서도 실제 데이터 흐름 기반의 분석이 가능해진다.