torch.Tensor.retain_grad

Tensor.retain_grad() → None

启用此 Tensor 在 backward() 期间填充其 grad。对于叶子张量，这是一个 no-op。