python实现nn.Tanh()
时间: 2024-03-21 16:42:25 浏览: 228
Python实现的NN神经网络算法完整示例
在 PyTorch 中,可以使用 `nn.Tanh()` 类来实现 `tanh` 激活函数。`nn.Tanh()` 类实现了 `tanh` 函数的前向传播计算,并且可以自动求导,方便后续的反向传播计算。
使用 `nn.Tanh()` 类的示例代码如下:
```python
import torch
import torch.nn as nn
x = torch.tensor([0.5, 1, 2, 3])
tanh = nn.Tanh()
y = tanh(x)
print(y)
```
在上述代码中,首先创建一个形状为 `(4,)` 的 PyTorch 张量 `x`,然后创建一个 `nn.Tanh()` 类的实例 `tanh`,并将 `x` 作为输入传递给 `tanh` 的 `__call__()` 方法,计算 `x` 的 `tanh` 值并将计算结果保存在变量 `y` 中,最后打印输出 `y` 的值。
输出结果如下:
```
tensor([0.4621, 0.7616, 0.9640, 0.9951])
```
在上述输出结果中,可以看到 `nn.Tanh()` 类计算出了 `x` 的 `tanh` 值,并将结果保存在了一个形状相同的 PyTorch 张量中。
阅读全文