Tanh

classtorch.nn.Tanh(*args, **kwargs)[源代码]

对每个元素应用双曲正切(Tanh)函数。

Tanh 被定义为:

$\text{Tanh}(x) = \tanh(x) = \frac{\exp(x) - \exp(-x)} {\exp(x) + \exp(-x)}$
形状:
  • 输入: $(*)$,其中$*$表示任意维度的数量。

  • 输出: $(*)$,形状与输入相同。

{BASE_RAW_UPLOAD_URL}/pytorch-doc-2.5/e63af8d38904626d01a0dcd358bad9b3.png

示例:

>>> m = nn.Tanh()
>>> input = torch.randn(2)
>>> output = m(input)
本页目录