pytorch의 배치 형태로 변경한다.
pytorch의 배치 형태로 변경한다.