若API未标明支持情况，则代表该API的支持情况待验证。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.autograd.Function |
||
torch.autograd.profiler.profile |
否 |
|
torch.autograd.profiler.emit_nvtx |
否 |
|
torch.autograd.detect_anomaly |
||
torch.autograd.set_detect_anomaly |
||
torch.autograd.graph.saved_tensors_hooks |
||
torch.autograd.graph.save_on_cpu |
||
torch.autograd.backward |
||
torch.autograd.grad |
||
torch.autograd.forward_ad.dual_level |
||
torch.autograd.forward_ad.make_dual |
||
torch.autograd.forward_ad.unpack_dual |
||
torch.autograd.functional.jacobian |
||
torch.autograd.functional.hessian |
||
torch.autograd.functional.vjp |
||
torch.autograd.functional.jvp |
||
torch.autograd.functional.vhp |
||
torch.autograd.functional.hvp |
||
torch.autograd.no_grad |
||
torch.autograd.enable_grad |
||
torch.autograd.set_grad_enabled |
||
torch.autograd.inference_mode |
||
torch.Tensor.grad |
||
torch.Tensor.requires_grad |
||
torch.Tensor.is_leaf |
||
torch.Tensor.backward |
||
torch.Tensor.detach |
||
torch.Tensor.detach_ |
||
torch.Tensor.register_hook |
||
torch.Tensor.retain_grad |
||
static Function.forward |
||
static Function.backward |
||
static Function.jvp |
||
FunctionCtx.mark_dirty |
||
FunctionCtx.mark_non_differentiable |
||
FunctionCtx.save_for_backward |
||
FunctionCtx.set_materialize_grads |
||
torch.autograd.gradcheck |
||
torch.autograd.gradgradcheck |
||
profile.export_chrome_trace |
否 |
|
profile.key_averages |
否 |
|
profile.total_average |
否 |
|
torch.autograd.profiler.load_nvprof |
否 |