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_

-