torch.no_grad() 上下文管理器怎么用

torch.no_grad() 上下文管理器可以用来禁用梯度计算，从而加速模型的推理过程。使用方法如下：

with torch.no_grad():
    # 在这里执行不需要梯度计算的代码

在这个上下文管理器中，所有的计算都不会被记录在计算图中，也不会对模型的参数进行更新。这个功能在测试模型时非常有用，因为测试时不需要计算梯度，只需要进行前向传播即可。