pytorch网络初始化权重

我的网络定义如下：

        self.net = nn.Sequential(
            nn.Linear(s_dim, 256),
            nn.Softplus(),
            nn.Linear(256, 256),
            nn.Softplus(),
            nn.Linear(256, a_dim)
        )

这是一个sequential快速方式，现在我想初始化每层的权重，使参数服从（0,1）正态分布，请教怎么操作？？

init