mindspore.mint.nn.SELU

class mindspore.mint.nn.SELU[源代码]

逐元素计算激活函数SELU（Scaled exponential Linear Unit）。

更多参考详见 mindspore.mint.nn.functional.selu()。

SELU函数图：

../../_images/SeLU.png
支持平台：

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)
>>> selu = mint.nn.SELU()
>>> output = selu(input)
>>> print(output)
[[-1.1113307 4.202804 -1.7575096]
[ 2.101402 -1.7462534 9.456309 ]]