快捷方式

dual_level

class torch.autograd.forward_ad.dual_level[source]

前向 AD 的上下文管理器,所有前向 AD 计算都必须在 dual_level 上下文中进行。

注意

dual_level 上下文适当地进入和退出双层以控制当前的前向 AD 层级,此 API 中其他函数默认使用该层级。

目前我们不打算支持嵌套 dual_level 上下文,因此只支持单个前向 AD 层级。要计算高阶前向梯度,可以使用 torch.func.jvp()

示例

>>> x = torch.tensor([1])
>>> x_t = torch.tensor([1])
>>> with dual_level():
...     inp = make_dual(x, x_t)
...     # Do computations with inp
...     out = your_fn(inp)
...     _, grad = unpack_dual(out)
>>> grad is None
False
>>> # After exiting the level, the grad is deleted
>>> _, grad_after = unpack_dual(out)
>>> grad is None
True

请参阅 前向模式 AD 教程,了解如何使用此 API 的详细步骤。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源