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