Tensor 연산에서 dimension을 지정하는 것에 관하여
HTML 삽입 미리보기할 수 없는 소스 numpy와 pytorch를 공부하면 자주 맞딱뜨리면서도 매번 헷갈리는 개념이 있습니다. 바로 행렬 연산에서 dimension을 지정하는 것인데요. 다음 예시 행렬을 보겠습니다: a = torch.arange(0, 18).view(3, 6) a >>> tensor([[ 0, 1, 2, 3, 4, 5], [ 6, 7, 8, 9, 10, 11], [12, 13, 14, 15, 16, 17]]) a는 $3 \times 6$ 행렬 또는 텐서입니다. 우리는 일반적으로 dim=0은 행을, dim=1을 열로 알고있고 실제로도 그렇습니다. 그렇다면, 위의 예시 행렬에서 행의 값들을 더해서 3개의 값을 반환 받고싶다면 어떻게 해야할까요? 행의 값들을 더하고 싶기 때문에 a.sum(..
2023. 12. 29.
