#pytorch

Torch.unsqueeze and squeeze

torch.unsqueeze 와 torch.squeeze 는 가끔 헷갈리는 개념이다. 머리로는 이해하는 것 같지만 막상 쓰려고 할때 팍 이해가 안가는 느낌이 많이 든다. 그래서 오늘 조금 정리해보려고 한다. Tensor 와 차원(Dimension) Pytorch 에서 Tensor 는 숫자값을 담는 다차원 배열이라고 생각하면 편하다. N 차원 텐서들을 아래와 같은 형태로 정의할 수 있다. zero_dim = torch.tensor(5) one_d...