!1024 add example for hsigmoid etc

Merge pull request !1024 from jiangjinsheng/nn_example
This commit is contained in:
mindspore-ci-bot 2020-05-09 15:58:48 +08:00 committed by Gitee
commit 08d86c483c
1 changed files with 21 additions and 1 deletions

View File

@ -122,6 +122,11 @@ class ELU(Cell):
Outputs:
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):
super(ELU, self).__init__()
@ -343,11 +348,16 @@ class PReLU(Cell):
w (float): The initial value of w. Default: 0.25.
Inputs:
- **input_data** (Tensor) - The input of Tanh.
- **input_data** (Tensor) - The input of PReLU.
Outputs:
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="")
def __init__(self, channel=1, w=0.25):
@ -394,6 +404,11 @@ class HSwish(Cell):
Outputs:
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):
super(HSwish, self).__init__()
@ -422,6 +437,11 @@ class HSigmoid(Cell):
Outputs:
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):
super(HSigmoid, self).__init__()