pytorch/torch/nn/parameter.pyi
Theo 4fac61a886 Fix typing on nn.Parameter (#25586)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/25399

As per https://github.com/pytorch/pytorch/issues/25580 I'm pushing this to test my changes on the CI
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25586

Differential Revision: D17259178

Pulled By: ezyang

fbshipit-source-id: d48cdd602bfda60c213f79a4f124df54a68ca698
2019-09-09 07:54:27 -07:00

8 lines
145 B
Python

from .. import Tensor
import builtins
class Parameter(Tensor):
def __init__(self, data: Tensor, requires_grad: builtins.bool): ...
...