mindspore.ops.tanhshrink

查看源文件
mindspore.ops.tanhshrink(input)[源代码]

逐元素计算输入tensor的Tanhshrink激活函数值。

Tanhshrink(x)=xTanh(x)
参数:
  • input (Tensor) - 输入tensor。

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> output = mindspore.ops.tanhshrink(mindspore.tensor([1., 2., 3., 2., 1.]))
>>> print(output)
[0.23840582 1.0359724  2.0049443  1.0359724  0.23840582]