tensor 차원 늘리기 & channel 추가
06 Feb 2020 | PyTorch만약 image가 (28,28)인 tensor면 앞에 차원을 늘려주기위해서는
image.unsqueeze(0)
를 해주면 (1,28,28) 이 된다.
이 때 channel을 3으로 늘리고 싶으면..
image = image.repeat(3, 1, 1)
를 해주면 결과적으로 (3,28,28)인 이미지가 만들어진다.
만약 image가 (28,28)인 tensor면 앞에 차원을 늘려주기위해서는
image.unsqueeze(0)
를 해주면 (1,28,28) 이 된다.
이 때 channel을 3으로 늘리고 싶으면..
image = image.repeat(3, 1, 1)
를 해주면 결과적으로 (3,28,28)인 이미지가 만들어진다.
Comments