torch.unsqueeze()와 torch.flatten()의 차이

1. 일반적인 torch array

• 코드

import torch
x = torch.linspace(-1,1,20)
x.shape

• 결과

2. torch.unsqueeze()

• 코드

import torch
x = torch.linspace(-1,1,20).unsqueeze(dim=1)
x.shape

• 결과
  

3. torch.flatten()

• 코드

x.flatten()
x.shape

• 결과