What is the LibTorch equivalent to PyTorch's torch.no_grad?

Question

When testing a network in PyTorch one can use with torch.no_grad():. What is the Libtorch (C++) equivalent? Thanks!

Oestriol

Answer 1

The equivalent in LibTorch is torch::NoGradGuard no_grad, see documentation.

Distefano

So I can just use it like this torch::NoGradGuard no_grad; and every following line operates with no grad? – Oestriol

Yes. It will affect the current scope you're in by disabling the gradient computation. – Distefano