본문으로 건너뛰기
재미있는 삽질 기록
Home
Categories
Home
Categories
#dim
1. dim 쉽게 이해하기
한참 Pytorch를 공부하고 있을 때, tensor.sum(dim=x) , torch.stack((tensors), dim=x) 등의 연산에서 dim=x 가 어떤 방향으로 연산이 이루어지는 건지 이해하기 어려웠었는데, 쉽게 이해할 수 있었던 방법을 정리해봤다. 단순한 규칙 알아두...
#pytorch #tensor #dim
|
2025-09-02