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