Python Keras partial custom activation function

Question

I have a NN with 6 output nodes and I want to use a custom activation function on the last three neurons of the layer. This is what I came up with:

def rounde