I need to add PReLU activation which has a learn-able parameter? What would be the best way to extend this library in this respect? thanks.