12345678910111213141516171819202122232425262728293031323334353637 |
- --- /usr/local/lib/python3.5/dist-packages/torch/nn/modules/activation.py
- +++ /usr/local/lib/python3.5/dist-packages/torch/nn/modules/activation.py
- @@ -37,9 +37,10 @@
- - Output: :math:`(N, *)`, same shape as the input
-
- Attributes:
- - weight (Tensor): the learnable weights of shape (:attr:`num_parameters`).
- + weight (Tensor): the learnable weights of shape (attr:`num_parameters`).
- + The attr:`dtype` is default to
-
- - .. image:: ../scripts/activation_images/PReLU.png
- + .. image:: scripts/activation_images/PReLU.png
-
- Examples::
-
- @@ -47,17 +48,16 @@
- >>> input = torch.randn(2)
- >>> output = m(input)
- """
- - __constants__ = ['num_parameters']
- - num_parameters: int
-
- - def __init__(self, num_parameters: int = 1, init: float = 0.25) -> None:
- + def __init__(self, num_parameters=1, init=0.25):
- self.num_parameters = num_parameters
- super(PReLU, self).__init__()
- self.weight = Parameter(torch.Tensor(num_parameters).fill_(init))
-
- - def forward(self, input: Tensor) -> Tensor:
- + @weak_script_method
- + def forward(self, input):
- return F.prelu(input, self.weight)
-
- - def extra_repr(self) -> str:
- + def extra_repr(self):
- return 'num_parameters={}'.format(self.num_parameters)
-
|