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函数图:
- 输入:
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]