12345678910111213141516171819202122 |
- --- /usr/local/lib/python3.5/dist-packages/torch/nn/modules/dropout.py
- +++ /usr/local/lib/python3.5/dist-packages/torch/nn/modules/dropout.py
- @@ -18,8 +18,8 @@
- inplace: If set to ``True``, will do this operation in-place. Default: ``False``
-
- Shape:
- - - Input: :math:`(*)`. Input can be of any shape
- - - Output: :math:`(*)`. Output is of the same shape as input
- + - Input: `Any`. Input can be of any shape
- + - Output: `Same`. Output is of the same shape as input
-
- Examples::
-
- @@ -31,6 +31,7 @@
- detectors: https://arxiv.org/abs/1207.0580
- """
-
- - def forward(self, input: Tensor) -> Tensor:
- + @weak_script_method
- + def forward(self, input):
- return F.dropout(input, self.p, self.training, self.inplace)
-
|