vilc
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix api bug about PReLU op.

This commit is contained in:
liuxiao93 2020-09-23 12:15:40 +08:00
parent 10e384674f
commit b3b59c30a6
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mindspore/ops/operations/nn_ops.py
View File

@ -2701,13 +2701,12 @@ class PReLU(PrimitiveWithInfer):
>>> weight = Tensor(np.array([0.1, 0.6, -0.3]), mindspore.float32)
>>> net = Net()
>>> result = net(input_x, weight)
[[[-0.1 1. ]
[ 0. 2. ]
[0. 0. ]]
[[-0.2 -0.1 ]
[2. -1.8000001]
[0.6 0.6 ]]]
[[[-0.1, 1.0],
[0.0, 2.0],
[0.0, 0.0]],
[[-0.2, -0.1],
[2.0, -1.8000001],
[0.6, 0.6]]]
"""
@prim_attr_register