E is just a linear function can be auto parametrised so Need not to worry about it . you can also use Belu from PyTorch