torch.autograd.forward_ad.unpack_dual¶
- torch.autograd.forward_ad.unpack_dual(tensor, *, level=None)[源代码]¶
解包“双张量”以获取其张量值和前向 AD 梯度。
结果是一个名为元组
(primal, tangent)
,其中primal
是tensor
的原张量的视图,而tangent
是tensor
的切线张量本身。这两个张量都不能是级别为level
的双张量。此函数是反向可微的。
示例
>>> with dual_level(): ... inp = make_dual(x, x_t) ... out = f(inp) ... y, jvp = unpack_dual(out) ... jvp = unpack_dual(out).tangent
请参阅 前向模式 AD 教程,了解有关如何使用此 API 的详细步骤。