torch_npu.optim
概述
torch_npu.optim.NpuFusedOptimizerBase
torch_npu.optim.NpuFusedSGD
torch_npu.optim.NpuFusedAdadelta
torch_npu.optim.NpuFusedLamb
torch_npu.optim.NpuFusedAdam
torch_npu.optim.NpuFusedAdamW
torch_npu.optim.NpuFusedAdamP
torch_npu.optim.NpuFusedBertAdam
torch_npu.optim.NpuFusedRMSprop
torch_npu.optim.NpuFusedRMSpropTF
父主题：
Ascend Extension for PyTorch自定义API