torch.nn导入init会报错？

问题；

        这个是报错的程序；

        这个是报错原因；

我的调查；

          我在pytorch官网上查到torch.nn与torch.nn.init不属于包含关系

总结；那我应该怎么修改源代码才不报错

直接 import torch.nn.init？