pytorch默认初始化_Pytorch参数初始化--默认与自定义

1. Pytorch module默认初始化

Pytorch 的Module都只带默认的初始化方法，而且初始化会调用此函数，因此我们定义好后，不用手动初始化。

Conv

class _ConvNd(Module):

__constants__ = ['stride', 'padding', 'dilation', 'groups', 'bias',

'padding_mode', 'output_padding', 'in_channels',

'out_channels', 'kernel_size']

def __init__(self, in_channels, out_channels, kernel_size, stride,

padding, dilation, transposed, output_padding,

groups, bias, padding_mode):

super(_ConvNd, self).__init__()

if in_channels % groups != 0:

raise ValueError('in_channels must be divisible by groups')

if out_channels % groups != 0:

raise ValueError('out_channels must be divisible by groups')

self.in_channels = in_channels

self.out_channels = out_channels

<