torch.Tensor.retain_grad

Tensor.retain_grad() → None

Включает возможность для этого тензора иметь заполненное значение grad во время вычисления backward(). Для лист-тензоров это действие ничего не делает.