nn.py 文件源码

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