若API“是否支持”为“是”，“限制与说明”为空，说明此API和原生API支持度保持一致。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.func.vmap |
否 |
|
torch.func.grad |
是 |
|
torch.func.grad_and_value |
是 |
|
torch.func.vjp |
否 |
|
torch.func.jvp |
否 |
|
torch.func.linearize |
是 |
|
torch.func.jacrev |
否 |
|
torch.func.jacfwd |
否 |
|
torch.func.hessian |
否 |
|
torch.func.functionalize |
否 |
|
torch.func.functional_call |
是 |
|
torch.func.stack_module_state |
是 |
|
torch.func.replace_all_batch_norm_modules_ |
否 |