mindspore.mint.nn.ReLU6
- class mindspore.mint.nn.ReLU6[源代码]
ReLU6激活函数。
更多参考详见
mindspore.mint.nn.functional.relu6()
。ReLU6激活函数图:
- 支持平台:
Ascend
样例:
>>> import mindspore >>> from mindspore import Tensor, mint >>> import numpy as np >>> input = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32) >>> relu6 = mint.nn.ReLU6() >>> output = relu6(input) >>> print(output) [[0. 4. 0.] [2. 0. 6.]]