Add energy constraint to batchnormalization layer

Question

I\'m studying the paper "An Introduction to Deep Learning for the Physical Layer", i should add an energy constraint ||X||^2 < 1 to batchnormal