Deep Learning/framework
[Pytorch] nn.ReflectionPad2d
개발챙
2021. 6. 16. 12:08
728x90
input tensor를 input boundary의 reflection(반사)를 이용해서 패딩 적용
2차원 아닌 N차원 패딩을 이용하려면 torch.nn.functional.pad()를 사용
- padding은 int형이나 tuple형 : 패딩 사이즈. int형이면 모든 바운더리에 같은 패딩을 적용. 4-tuple이면 (padding_left, padding_right, padding_top, padding_bottom)으로 적용
- Input: (N, C, H_in, W_in)
- Output: (N, C, H_out, W_out)
- W_out = W_in + padding_left + padding_right
https://pytorch.org/docs/stable/generated/torch.nn.ReflectionPad2d.html
728x90