I would like to regularize the layers of a CNN with

|(W^T * W - I)|

How can I do that in Keras?