| @@ -122,6 +122,11 @@ class ELU(Cell): | |||||
| Outputs: | Outputs: | ||||
| Tensor, with the same type and shape as the `input_data`. | Tensor, with the same type and shape as the `input_data`. | ||||
| Examples: | |||||
| >>> input_x = Tensor(np.array([-1, -2, 0, 2, 1]), mindspore.float32) | |||||
| >>> elu = nn.ELU() | |||||
| >>> elu(input_x) | |||||
| """ | """ | ||||
| def __init__(self, alpha=1.0): | def __init__(self, alpha=1.0): | ||||
| super(ELU, self).__init__() | super(ELU, self).__init__() | ||||
| @@ -343,11 +348,16 @@ class PReLU(Cell): | |||||
| w (float): The initial value of w. Default: 0.25. | w (float): The initial value of w. Default: 0.25. | ||||
| Inputs: | Inputs: | ||||
| - **input_data** (Tensor) - The input of Tanh. | |||||
| - **input_data** (Tensor) - The input of PReLU. | |||||
| Outputs: | Outputs: | ||||
| Tensor, with the same type and shape as the `input_data`. | Tensor, with the same type and shape as the `input_data`. | ||||
| Examples: | |||||
| >>> input_x = Tensor(np.array([-1, -2, 0, 2, 1]), mindspore.float32) | |||||
| >>> prelu = nn.PReLU() | |||||
| >>> prelu(input_x) | |||||
| """ | """ | ||||
| @cell_attr_register(attrs="") | @cell_attr_register(attrs="") | ||||
| def __init__(self, channel=1, w=0.25): | def __init__(self, channel=1, w=0.25): | ||||
| @@ -394,6 +404,11 @@ class HSwish(Cell): | |||||
| Outputs: | Outputs: | ||||
| Tensor, with the same type and shape as the `input_data`. | Tensor, with the same type and shape as the `input_data`. | ||||
| Examples: | |||||
| >>> input_x = Tensor(np.array([-1, -2, 0, 2, 1]), mindspore.float16) | |||||
| >>> hswish = nn.HSwish() | |||||
| >>> hswish(input_x) | |||||
| """ | """ | ||||
| def __init__(self): | def __init__(self): | ||||
| super(HSwish, self).__init__() | super(HSwish, self).__init__() | ||||
| @@ -422,6 +437,11 @@ class HSigmoid(Cell): | |||||
| Outputs: | Outputs: | ||||
| Tensor, with the same type and shape as the `input_data`. | Tensor, with the same type and shape as the `input_data`. | ||||
| Examples: | |||||
| >>> input_x = Tensor(np.array([-1, -2, 0, 2, 1]), mindspore.float16) | |||||
| >>> hsigmoid = nn.HSigmoid() | |||||
| >>> hsigmoid(input_x) | |||||
| """ | """ | ||||
| def __init__(self): | def __init__(self): | ||||
| super(HSigmoid, self).__init__() | super(HSigmoid, self).__init__() | ||||