pytorch报错：module 'torch.nn' has no attribute 'init'

在调用pytorch的init模块的时候，出现错误：

module 'torch.nn' has no attribute 'init'

但是在github中看到是有这个文件的，所以再看了下本地安装路径，也有这个文件。想到应该是没有import的原因，就把文件~/anaconda3/lib/python3.6/site-packages/torch/nn/init.py中的代码增加一行：

from .modules import *
from .parameter import Parameter
from .parallel import DataParallel
from .init import *

问题解决。
ps：这在python2中是没有这个问题的，python3对于import当前路径的规则有些改变。