def forward(self, input):
out = self.convs(input)
linear = self.fc(out.view(-1, self.fc_dim))
return F.sigmoid(linear)
微信
公众号
扫码关注公众号