擴充套件#

本節內容將深入探討如何擴充套件 PyTorch 的功能。內容涵蓋自定義操作、前端 API 以及 C++ 擴充套件和排程器使用等高階主題。

這是關於 PyTorch 中自定義運算子所有內容的著陸頁。

擴充套件 PyTorch,前端 API,C++,CUDA

在 Python 中建立自定義運算子。適用於將 Python 函式黑盒化以供 torch.compile 使用。

擴充套件 PyTorch,前端 API,C++,CUDA

如何使用自定義 C++ 和 CUDA 運算子擴充套件 PyTorch。

擴充套件 PyTorch,前端 API,C++,CUDA

學習如何編寫支援二階導數的自定義 autograd Function。

擴充套件 PyTorch,前端 API

學習如何建立自定義 autograd Function，將 Batch Norm 融合到卷積中以提高記憶體使用率。

擴充套件 PyTorch,前端 API

排程器是 PyTorch 的內部元件，負責確定呼叫 torch::add 等函式時實際執行的程式碼。

擴充套件 PyTorch,前端 API,C++

學習如何擴充套件排程器以新增位於 pytorch/pytorch 倉庫之外的新裝置，並對其進行維護，以保持與原生 PyTorch 裝置同步。

擴充套件 PyTorch,前端 API,C++

學習如何整合一個位於 pytorch/pytorch 倉庫之外的新後端，並對其進行維護，以保持與原生 PyTorch 後端同步。

擴充套件 PyTorch,前端 API,C++

文件