def __init__(self,inp,outp): super(ActorNet, self).__init__() self.in_to_y1=nn.Linear(inp,512) self.in_to_y1.weight.data.normal_(0,0.1) self.y1_to_y2=nn.Linear(512,256) self.y1_to_y2.weight.data.normal_(0,0.1) self.out=nn.Linear(256,outp) self.out.weight.data.normal_(0,0.1)
时间: 2023-05-21 10:01:49 浏览: 167
这是一个 PyTorch 中的神经网络模型的初始化函数,其中包括三个线性层,分别是输入层到第一隐藏层、第一隐藏层到第二隐藏层、第二隐藏层到输出层。每个线性层都有一个权重矩阵,使用正态分布初始化。其中,inp 表示输入的特征数,outp 表示输出的特征数。
阅读全文