开发者
资源

实现函数化转换

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