概述

提供PyTorch2.4.0/2.3.1/2.1.0版本原生API接口在昇腾设备上的支持情况与限制说明，以及Ascend Extension for PyTorch自定义API的接口原型、功能描述、参数说明与调用示例等。

PyTorch原生API接口使用方法请参考PyTorch Documentation。

Ascend Extension for PyTorch提供的接口均遵循PyTorch社区公开接口规范，具体规范可参考Public API definition and documentation。本文档展示的接口为Ascend Extension for PyTorch的对外公开接口，其他内部接口后续版本可能会被修改或删除，请尽量不要使用，若必须使用，请在昇腾社区（https://gitee.com/ascend/pytorch/issues）提交issue获取帮助。

Ascend Extension for PyTorch使用monkey-patch方式与PyTorch接口进行对接，即将Ascend Extension for PyTorch部分接口动态替换至PyTorch接口中，使用户在昇腾NPU上可以继续使用熟悉的PyTorch接口。

Ascend Extension for PyTorch项目采用C++和python联合开发，当前正式对外接口仅包括python接口。C++接口为内部使用接口，不建议用户使用。

当前有部分接口会被标记为beta类接口。beta类接口属于实验性接口，部分场景下可能出现异常，请谨慎使用此类接口。我们致力于将beta类接口纳入稳定接口中，但完成之前仍有可能会根据需要进行改动，包括但不限于参数变更、名称变更和移除。