xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix the examples of P.ReLU6

This commit is contained in:
万万没想到 2020-04-01 22:16:29 +08:00 committed by Gitee
parent 0830ad932b
commit 4feba6332d
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -190,7 +190,6 @@ class ReLU6(PrimitiveWithInfer):
>>> input_x = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]], np.float32)) >>> input_x = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]], np.float32))
>>> relu6 = ReLU6() >>> relu6 = ReLU6()
>>> result = relu6(input_x) >>> result = relu6(input_x)
>>> assert result.asnumpy() == Tensor(np.array([[0, 4.0, 0.0], [2.0, 0.0, 6.0]], np.float32)).asnumpy()
""" """
@prim_attr_register @prim_attr_register