배치사이즈 채널 너비 높이

안녕하세요!

일반적인 이미지를 파이썬 내로 불러오면 기본적으로 채널 수가 가장 뒤에 있습니다. 지금 가지고 계신 컬러 이미지를 아무거나 한 장을 불러 오시면 크기가 너비x높이x채널수로 되어 있는 것을 확인하실 수 있을겁니다. 그 예시로 train_images를 만든 것입니다. 그런데 파이토치는 이미지 크기가 채널수x너비x높이 형태가 필요하니 크기를 (너비x높이x채널수)에서 (채널수x너비x높이)로 변환해야한다는 내용입니다 :)