nn.py 文件源码

def __init__(self, input_dim, output_dim, dropout=0, softplus_boost=1.0):
        super(ProposalMultivariateNormal, self).__init__()
        self.mean_lin1 = nn.Linear(input_dim, input_dim)
        self.mean_drop = nn.Dropout(dropout)
        self.mean_lin2 = nn.Linear(input_dim, output_dim)

        self.vars_lin1 = nn.Linear(input_dim, input_dim)
        self.vars_drop = nn.Dropout(dropout)
        self.vars_lin2 = nn.Linear(input_dim, output_dim)

        self.softplus_boost = softplus_boost

        init.xavier_uniform(self.mean_lin1.weight, gain=init.calculate_gain('relu'))
        init.xavier_uniform(self.mean_lin2.weight)
        init.xavier_uniform(self.vars_lin1.weight, gain=init.calculate_gain('relu'))
        init.xavier_uniform(self.vars_lin2.weight)