Contents Introduction Short intro to Python extension objects in C/C++ Zero-copy PyTorch Tensor to Numpy and vice-versa Tensor Storage Shared Memory DLPack: a hope for the Deep Learning frameworks Babel Introduction This post is a tour around the PyTorch codebase, it is meant to be a guide for the architectural design
Source: PyTorch – Internal Architecture Tour | Terra Incognita