mindspore.mint.nn.SELU
- class mindspore.mint.nn.SELU[源代码]
逐元素计算SELU(缩放指数线性单元)激活函数。
更多参考详见
mindspore.mint.nn.functional.selu()。SELU函数图:
- 支持平台:
Ascend
样例:
>>> import mindspore >>> selu = mindspore.mint.nn.SELU() >>> input_tensor = mindspore.tensor([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]], mindspore.float32) >>> output = selu(input_tensor) >>> print(output) [[-1.1113307 4.202804 -1.7575096] [ 2.101402 -1.7462534 9.456309 ]]