mindspore.mint.nn.Tanh

查看源文件
class mindspore.mint.nn.Tanh[源代码]

逐元素计算Tanh(双曲正切值)激活函数。

Tanh函数定义为:

\[tanh(x_i) = \frac{\exp(x_i) - \exp(-x_i)}{\exp(x_i) + \exp(-x_i)} = \frac{\exp(2x_i) - 1}{\exp(2x_i) + 1},\]

其中 \(x_i\) 是输入Tensor的元素。

Tanh函数图:

../../_images/Tanh.png
输入:
  • input (Tensor) - 输入tensor。

输出:

Tensor。

支持平台:

Ascend

样例:

>>> import mindspore
>>> input = mindspore.tensor([1, 2, 3, 2, 1], mindspore.float16)
>>> tanh = mindspore.mint.nn.Tanh()
>>> output = tanh(input)
>>> print(output)
[0.7617 0.964  0.995  0.964  0.7617]