tensorplay.autograd.function.once_differentiable

tensorplay.autograd.function.once_differentiable(fn)[source]

Decorator to make a custom autograd Function’s backward run once, with gradients detached and grad-mode disabled inside.