!89 fix the examples of P.ReLU6
Merge pull request !89 from 万万没想到/I1DBUG
This commit is contained in:
commit
34b74e4430
|
@ -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))
|
||||
>>> relu6 = ReLU6()
|
||||
>>> 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
|
||||
|
|
Loading…
Reference in New Issue