본문으로 바로가기

tensor 차원 늘리기 & channel 추가

category AI/Tip 2020. 2. 6. 20:07
336x280(권장), 300x250(권장), 250x250, 200x200 크기의 광고 코드만 넣을 수 있습니다.

 

 

 

 

 

만약 image가 (28,28)인 tensor면 앞에 차원을 늘려주기위해서는

image.unsqueeze(0)

를 해주면 (1,28,28) 이 된다.

 

 

이 때 channel을 3으로 늘리고 싶으면..

image = image.repeat(3, 1, 1)

를 해주면 결과적으로 (3,28,28)인 이미지가 만들어진다.