[Python] PyTorch 텐서 차원 다루기
딥러닝에서는 차원(Dimension) 개념이 굉장히 중요합니다.배치 크기(batch_size), 채널(channel), 높이(height), 너비(width) 같은 정보들이 모두 차원으로 표현되기 때문이죠.이번 포스팅에서는 파이토치(PyTorch)에서 차원을 다루는 기본 스킬을 예제로 정리해보겠습니다. 전체 코드입니다. import torch# 1. 텐서(Tensor) 만들기# -----------------------------# 1차원 벡터 (길이 4)a = torch.tensor([1, 2, 3, 4])print("a:", a)print("a.shape:", a.shape) # torch.Size([4])# 2차원 행렬 (2행 3열)b = torch.tensor([[1, 2, 3], [4, 5, ..
2025. 8. 26.