使用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 |
错误消失。
本文最后更新于2019年5月21日,已超过 1 年没有更新,如果文章内容或图片资源失效,请留言反馈,我们会及时处理,谢谢!