[1차원 추가 및 제거] Pytorch squeeze / unsqueeze
상황: 딥러닝을 하다보면 1차원을 추가 및 제거를 위한 단계가 필요하게 된다. 1. squeeze : 1차원 제거 역할. import torch x= torch.ones(5,4,1,4,1) x1 = x.squeeze() # 모든 1차원 제거 print(x1.size()) # torch.Size([5, 4, 4]) x2= x.squeeze(dim = 2) print(x2.size()) # torch.Size([5, 4, 4, 1]) x3= x.squeeze(dim = -1) # dim=4 와 같음 print(x3.size()) # torch.Size([5, 4, 1, 4]) x4= x.squeeze(dim = 1) # 잘못된 차원 삭제 불가능. print(x4.size()) # torch.Size([5, ..
2023. 12. 23.