diff --git a/torch/_C/__init__.pyi.in b/torch/_C/__init__.pyi.in index 20a2244a76878..8b0886a360de1 100644 --- a/torch/_C/__init__.pyi.in +++ b/torch/_C/__init__.pyi.in @@ -1455,6 +1455,7 @@ class _TensorMeta(type): ... # Defined in torch/csrc/autograd/python_variable.cpp class _TensorBase(metaclass=_TensorMeta): requires_grad: _bool + retains_grad: _bool shape: Size data: Tensor names: List[str]