若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_ |