torch.Tensor.retains_grad

Tensor.retains_grad

Является True, если этот тензор является не-листом, а его grad включён для заполнения во время backward(), False в противном случае.