torch.utils.checkpoint 若API“是否支持”为“是”,“限制与说明”为“-”,说明此API和原生API支持度保持一致。 API名称 是否支持 限制与说明 torch.utils.checkpoint.checkpoint 是 - torch.utils.checkpoint.checkpoint_sequential 是 支持fp32 父主题: PyTorch 2.6.0版本