In a torch.autograd.Function class, the model loaded with torch.load cannot get a result with (requires_grad = True)

Duplicate of Grad is None even when I set requires_grad=True
Answered there.