若API未标明支持情况，则代表该API的支持情况待验证。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.autograd.Function |
||
torch.autograd.profiler.profile |
否 |
|
torch.autograd.profiler.emit_nvtx |
否 |
|
torch.autograd.profiler.emit_itt |
否 |
|
torch.autograd.detect_anomaly |
||
torch.autograd.set_detect_anomaly |
||
torch.autograd.graph.saved_tensors_hooks |
||
torch.autograd.graph.save_on_cpu |
||
torch.autograd.graph.disable_saved_tensors_hooks |
否 |
|
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 |
否 |
|
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 |
否 |