torch.func

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

父主题： PyTorch原生API支持度