This PR includes the new design pattern for Linear, Conv2d, and AvgPooling. It supports both CPU and CUDA devices. In addition, we implement a python bindings of C++ code.
Changes Made
Linear layer for both CPU and CUDA
Conv2d Layer for both CPU and CUDA
AvgPool2d Layer for both CPU and CUDA
Python bindings of C++ code (see folder pytagi/pytagi_v1)
Description
This PR includes the new design pattern for Linear, Conv2d, and AvgPooling. It supports both CPU and CUDA devices. In addition, we implement a python bindings of C++ code.
Changes Made