@dynamic_spec으로 배치 가변 차원을 한곳에 — DAKER에서 자세히 보기

2026.09.15 08:32 63 조회

PyTorch 2.14의 @dynamic_spec과 ShapesSpec은 가변 배치·시퀀스 차원을 함수나 모듈에 붙여 둡니다. torch.compile과 torch.export가 같은 선언을 읽어, 호출부마다 다른 dynamic 설정을 흩뿌리지 않아도 됩니다.


• 가변으로 둘 차원을 먼저 고르고 스펙 한곳에만 적습니다

• compile과 export가 같은 규칙을 따릅니다

• 호출부 dynamic_shapes만 쓰는 방식과 비교해 보시면 됩니다


원문: https://daker.ai/community/pytorch-2-14-dynamic-spec-compile-export-batch-dim