with_enable_grad {torch} | R Documentation |
Context-manager that enables gradient calculation. Enables gradient calculation, if it has been disabled via with_no_grad.
with_enable_grad(code)
local_enable_grad(.env = parent.frame())
code |
code to be executed with gradient recording. |
.env |
The environment to use for scoping. |
This context manager is thread local; it will not affect computation in other threads.
local_enable_grad()
: Locally enable gradient computations.
if (torch_is_installed()) {
x <- torch_tensor(1, requires_grad = TRUE)
with_no_grad({
with_enable_grad({
y <- x * 2
})
})
y$backward()
x$grad
}