昇腾社区首页
中文
注册
开发者
下载

实现函数化转换

TorchAir已经支持了社区在PyTorch 2.6+版本提供的自动函数化能力,您无需再手动实现函数化。自动函数化的实现可以简单理解为先对要被修改的输入Tensor进行一次拷贝,使用拷贝后的Tensor调用In-place算子,后续使用这些输入的地方也将被替换为拷贝后的Tensor。