def forward(self,x):
    x = self.main(x)
    output = x.clamp(-5,5).torch.sigmoid()*scale
    return output