torch.Tensor.retains_grad

Tensor.retains_grad

如果此张量是非叶子张量且其 grad 在 backward() 期间被启用以填充，则为 True，否则为 False。