代码解释: 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 18:01:49 浏览: 174
ABAQUS二节点Truss_UEL_uel abaqus
5星 · 资源好评率100%
这是一个神经网络的初始化函数,其中inp是输入的特征数,outp是输出的特征数。该神经网络包含三个全连接层,分别是输入层到第一隐藏层的线性层(self.in_to_y1)、第一隐藏层到第二隐藏层的线性层(self.y1_to_y2)和第二隐藏层到输出层的线性层(self.out)。在初始化时,权重数据使用正态分布随机初始化,均值为0,标准差为0.1。
阅读全文