pytorch有哪些激活函数层
时间: 2023-11-10 17:42:20 浏览: 135
Pytorch中的激活函数
在PyTorch中,常见的激活函数层有以下几种:
1. ReLU(Rectified Linear Unit)激活函数层:torch.nn.ReLU()
2. Sigmoid激活函数层:torch.nn.Sigmoid()
3. Tanh激活函数层:torch.nn.Tanh()
4. Softmax激活函数层:torch.nn.Softmax(dim=n)
其中,n为需要进行softmax计算的维度。
阅读全文