使用pytorch的时候,如果我们使用torch.nn.functional中的sigmoid和tanh函数的话
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
可能会报错:
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里,因此呢,需要修改一下:
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 年没有更新,如果文章内容或图片资源失效,请留言反馈,我们会及时处理,谢谢!