nn.py 文件源码

def __init__(self, input_dim, mixture_components=10, dropout=0):
        super(ProposalUniformContinuousAlt, self).__init__()
        self.mixture_components = mixture_components
        self.output_dim = 3 * mixture_components
        self.lin1 = nn.Linear(input_dim, input_dim)
        self.lin2 = nn.Linear(input_dim, self.output_dim)
        self.drop = nn.Dropout(dropout)
        init.xavier_uniform(self.lin1.weight, gain=init.calculate_gain('relu'))
        init.xavier_uniform(self.lin2.weight)