扩展#

本节提供了关于扩展 PyTorch 功能的见解。它涵盖了自定义操作、前端 API 以及 C++ 扩展和调度器使用等高级主题。

这是关于 PyTorch 中自定义操作器的所有内容的着陆页。

扩展-PyTorch,前端-APIs,C++,CUDA

在 Python 中创建自定义操作器。对于使用 torch.compile 的 Python 函数黑盒化很有用。

扩展-PyTorch,前端-APIs,C++,CUDA

如何使用自定义 C++ 和 CUDA 操作器扩展 PyTorch。

扩展-PyTorch,前端-APIs,C++,CUDA

学习如何编写支持双向反向传播的自定义 autograd 函数。

扩展-PyTorch,前端-APIs

学习如何创建一个自定义 autograd 函数，将批归一化融合到卷积中，以提高内存使用率。

扩展-PyTorch,前端-APIs

调度器是 PyTorch 的一个内部组件，负责确定调用像 torch::add 这样的函数时应该实际运行哪些代码。

扩展-PyTorch,前端-APIs,C++

学习如何扩展调度器以添加一个新的设备，该设备存在于 pytorch/pytorch 仓库之外，并对其进行维护以使其与原生 PyTorch 设备保持同步。

扩展-PyTorch,前端-APIs,C++

学习如何集成一个新的后端，该后端存在于 pytorch/pytorch 仓库之外，并对其进行维护以使其与原生 PyTorch 后端保持同步。

扩展-PyTorch,前端-APIs,C++

文档