torch_npu.optim.NpuFusedOptimizerBase

torch_npu.optim.NpuFusedOptimizerBase(params)

通过张量融合实现优化器的基础类,实现梯度清零、梯度更新等优化器基本功能,用户可进行继承实现自定义融合优化器。