pychram+python 看源码
打开pychram , 按住crtl ,用鼠标点击你想要的函数,pycharm会自动打开所选的函数的源码。
另外一个 快捷方式:找目标。crtl+F
结合crtl+F快捷键,键入搜索目标关键字,能帮助你更快找到目标。
这里,看到了Pytorch 的Module都只带默认的初始化方法,而且初始化会调用此函数,因此我们定义好后,不用手动初始化。
在__init__中最后一行调用函数reset_parameters进行参数初始化,卷积函数都继承了_ConvNd,因此所有卷积module都自动初始化。
nn.conv2d的初始化方法为kaiming_uniform_ 。