Dropout 丢弃函数的使用

Class USeDropout(nn.Module):
 
    def __init__(self):
 
        super(DropoutFC, self).__init__()
 
        self.fc = nn.Linear(100,20)
 
        self.dropout = nn.Dropout(p=0.5)#声明
 
 
    def forward(self, input):
 
        out = self.fc(input)
 
        out = self.dropout(out)#定义调用
 
        return out
 
Net = USeDropout()
 
Net.train()

由上述代码知道，是在全连接层后使用，声明变量的位置无所谓，但是使用要在全连接层之后。