torch.func
Note
若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_ | 否 | - |