기술/PyTorch
nn.AdaptiveAvgPool1d
용나리
2022. 2. 10. 14:54
https://pytorch.org/docs/stable/generated/torch.nn.AdaptiveAvgPool1d.html
AdaptiveAvgPool1d — PyTorch 1.10 documentation
Shortcuts
pytorch.org
이미지 사이즈를 torch.nn.AdaptiveAvgPool1d(output_size) 의 output_size로 맞춘다. 물론 여기선 1d인거고, 2d, 3d도 마찬가지일거임.