torch.autograd.forward_ad.unpack_dual

torch.autograd.forward_ad.unpack_dual(tensor, *, level=None)

解包一个“双重张量”以获取其张量值和前向自动微分梯度。

结果是一个命名元组 (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

请参阅前向模式自动微分教程 以了解如何使用此API的详细步骤。