使用pytorch的时候,如果我们使用torch.nn.functional中的sigmoid和tanh函数的话
|
1 2 3 4 |
y_relu = F.relu(x).data.numpy y_sigmod = F.sigmod(x).data.numpy y_tanh = F.tanh(x).data.numpy y_softplus = F.softplus(x).data.numpy |
可能会报错:
|
1 2 3 4 5 6 7 8 |
AttributeError Traceback (most recent call last) <ipython-input-65-1e4b47833ef8> in <module> 1 y_relu = F.relu(x).data.numpy ----> 2 y_sigmod = F.sigmod(x).data.numpy 3 y_tanh = F.tanh(x).data.numpy 4 y_softplus = F.softplus(x).data.numpy AttributeError: module 'torch.nn.functional' has no attribute 'sigmod' |
这是因为新版本中的这两个激活函数已经放到了torch里,因此呢,需要修改一下:
|
1 2 3 4 |
y_relu = F.relu(x).data.numpy y_sigmod = torch.sigmoid(x).data.numpy y_tanh = torch.tanh(x).data.numpy y_softplus = F.softplus(x).data.numpy |
错误消失。





最新评论
哈哈
同样
站长您好,亚马逊云咨询推广资源,望建立联系,可邮件,谢谢。
换友情链接吗?
看你的站做的挺不错的
恭喜!!太强了,硕博连读啊
雁过留毛,人过留名。
看不懂但大受震撼