[Pytorch] nn.ReflectionPad2d
2021. 6. 16. 12:08ㆍDeep Learning/framework
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
'Deep Learning > framework' 카테고리의 다른 글
| [Pytorch] Variable (0) | 2021.06.19 |
|---|---|
| [pytorch] nn.Upsample (0) | 2021.06.16 |
| [Pytorch] Initializer (torch.nn.init) (0) | 2021.06.15 |
| [Pytorch] Tensor 의 .view()메소드 / .reshape()메소드 (0) | 2021.05.24 |
| [Pytorch] torch.sort (0) | 2021.05.22 |