mindspore.nn.Tanh

class mindspore.nn.Tanh[源代码]

Tanh激活函数。

按元素计算Tanh函数,返回一个新的Tensor,该Tensor是输入元素的双曲正切值。

Tanh函数定义为:

tanh(xi)=exp(xi)exp(xi)exp(xi)+exp(xi)=exp(2xi)1exp(2xi)+1,

其中 xi 是输入Tensor的元素。

输入:

  • x (Tensor) - 任意维度的Tensor,数据类型为float16或float32的输入。

输出:

Tensor,数据类型和shape与 x 的相同。

异常:

  • TypeError - x 的数据类型既不是float16也不是float32。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([1, 2, 3, 2, 1]), mindspore.float16)
>>> tanh = nn.Tanh()
>>> output = tanh(x)
>>> print(output)
[0.7617 0.964  0.995  0.964  0.7617]