LibTorch 相当于 PyTorch 的 torch.no_grad 是什么?

MD9*_*D98 8 c++ python pytorch autograd libtorch

在 PyTorch 中测试网络时,可以使用with torch.no_grad():. Libtorch (C++) 的等效项是什么?谢谢!

Iva*_*van 9

LibTorch 中的等效项是torch::NoGradGuard no_grad请参阅文档

  • 是的。它将通过禁用梯度计算来影响您当前所处的范围。 (3认同)