nn.py 文件源码

def __init__(self, input_dim, output_dim, dropout=0, softmax_boost=1.0):
        super(ProposalUniformDiscrete, self).__init__()
        self.lin1 = nn.Linear(input_dim, input_dim)
        self.lin2 = nn.Linear(input_dim, output_dim)
        self.drop = nn.Dropout(dropout)
        self.softmax_boost = softmax_boost
        init.xavier_uniform(self.lin1.weight, gain=init.calculate_gain('relu'))
        init.xavier_uniform(self.lin2.weight)