·
Follow
Published in
See Also
What is GPU programming | Red Hat Developer
ITNEXT
·
7 min read
·
Aug 2, 2024
--
So far we have created all building blocks (auto-differentiation, loss functions, layers, datasets, transforms). In order to efficiently process images we need layers which are designed for image processing specifically. Here we introduce and implement one of such layers: 2D convolution.
All the source code with samples is available on Github (under dark 5).