I want to be able to wrap all convolutional parameters (w) of my model with a function e.g. softmax(w), sigmoid(w), or alpha*w, etc. Is there any way of making this function part of the graph using hooks etc.?