torch.nn.Unfold 예제1 [Pytorch] torch.nn.Unfold torch.nn.Unfold 는 다음과 같다. batched tensor에 대하여 마치 convolution 처럼 sliding 움직이면서 그 local block을 구하는 것이다. 다음 사이트에서 자세한 설명을 볼 수 있다. https://pytorch.org/docs/stable/generated/torch.nn.Unfold.html Unfold — PyTorch 2.0 documentation Shortcuts pytorch.org 예를들면 다음과 같다. MNIST train 1번 그림인데, 여기에 Unfold 를 kernel=7, stride=7 padding=0 으로 주면 오른쪽 그림과 같이 kernel 기준으로 나뉘게 되고 하나의 local block 은 1(=c) x (kernel_w *ke.. 2023. 3. 16. 이전 1 다음