马春杰杰 Exit Reader Mode

[mcj]AttributeError: module ‘torch.nn.functional’ has no attribute ‘sigmod’

使用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 年没有更新,如果文章内容或图片资源失效,请留言反馈,我们会及时处理,谢谢!