一、内置随机化 init函数
二、内置常数化 init函数
三、不同层 or 块进行不同的init
四、自定义init函数
解释
1、apply(my_init)之后,神经网络会将每一层的权重和偏置分别放入my_init()函数并将结果赋予它们自身
2、my_init()函数接收每一层的权重和偏置,并采用torch的随机初始化(这是我随便写的,可以根据需求改写)
3、net.weight / net.bias的数据类型是torch.nn.parameter.Parameter,需要用.data将其转换为torch.tensor才能进行赋值
4、net.weight / net.bias的数据类型是torch.nn.parameter.Parameter,也可以用torch.nn.Parameter直接进行初始化,而不需要转换