torch-nn-functional

PyTorch에서 저수준 텐서 연산과 신경망 관련 함수들을 제공하는 모듈로서 연산을 직접 호출해 커스텀 변환을 구현할 때 사용된다. torch.nn.functional의 연산은 GPU에서 브로드캐스트·배치 연산으로 동작하므로 반복적인 Python 루프 대신 텐서 연산만으로 레터박스와 리사이즈를 처리하면 병렬 처리 이점을 얻을 수 있다. 이 글에서는 Aspect-Pad가 torch.nn.functional만 사용해 완전한 GPU 배치 전처리를 달성했다고 보고되었다.