torch.Tensor.retains_grad

Tensor.retains_grad

如果此张量是非叶节点且其 grad 在 backward() 时启用以填充，则返回 True ，否则返回 False 。