快捷方式

torch.autograd.forward_ad.unpack_dual

torch.autograd.forward_ad.unpack_dual(tensor, *, level=None)[源代码]

解包“双张量”以获取其张量值和前向 AD 梯度。

结果是一个名为元组 (primal, tangent),其中 primaltensor 的原张量的视图,而 tangenttensor 的切线张量本身。这两个张量都不能是级别为 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 的详细步骤。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源