[PyTorch] (B,C,H,W) 이어 붙이기 tensor.randn((B,) + (C,H,W))

 

CNN을 이용하여 Image Classification하는 훈련 train을 시킬 때 데이터를 (배치사이즈(B), 채널(C), 높이(H), 너비(W))를 차례대로 텐서의 크기를 맞춰서 코드를 작성해야할 때가 있다.

 

이때 배치사이즈만 따로 떨어져 있는 때가 있다.

이 때 tensor.randn((B,) + (C,H,W))를 이용하면 tensor((B,C,H,W)) 사이즈라는 것을 알릴 수 있다.

 

예제

# x_dim = (1,28,28)
batch_in.view(-1,x_dim)
#(batch_size, 1 * 28 * 28)

batch_in.view((-1,)+x_dim)
#(batch_size, 1, 28, 28)