kingmandooさんの投稿

Q＆A

ToTensor와 DataLoader관련질문드립니다!

이 질문의 답에 대한 질문이 있어서 답글 답니다. 3번 질문의 답 중에서 "하지만 이미지 데이터는 기본적으로 다차원의 텐서입니다. 텐서의 크기는 (Batch size, Number of Channels, Height, Width)이기 때문에 (Number of Channels, Height, Width)에 해당되는 차원들을 하나의 Hidden Feature로 "squeeze"하여 (Batch size, Hidden Feature Size) 크기로 맞춰주는 것입니다." 라고 답을 주셨는데, 하나의 차원으로 "squeeze"하는 것은 flatten() 함수이지 않나요? 실제로 다음과 같이 실행해도 결과에 변화가 없어서 질문 드렸습니다. img = train_features[0].squeeze() # torch.Size([3, 32, 32]) 출력 img = train_features[0] # torch.Size([3, 32, 32]) 출력