Another approach would be to apply some kind of normalisation in the forward pass, but that feels like a hacky approach when all I need to do is constrain the values in a tensor. I don’t want to add extra complexity to the graph.