mindspore.ops.tanhshrink

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

Tanhshrink激活函数, \(Tanhshrink(x)=x-Tanh(x)\) ,其中 \(x\) 即输入 input

详情请查看 mindspore.nn.Tanhshrink

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore as ms
>>> import mindspore.ops as ops
>>> from mindspore import Tensor
>>> import numpy as np
>>> input = Tensor(np.array([1, 2, 3, 2, 1]), ms.float16)
>>> output = ops.tanhshrink(input)
>>> print(output)
[0.2383 1.036  2.004  1.036  0.2383]