You could directly return the desired activation from the forward method or you could use forward hooks as described here.
forward