torch.autograd.function.FunctionCtx.set\_materialize\_grads =========================================================== .. currentmodule:: torch.autograd.function .. automethod:: FunctionCtx.set_materialize_grads