Squared modulus layer. Returns a keras layer that implements a squared modulus operator. To implement the squared modulus of C complex-valued channels, the expected input dimension is N*1*W*(2*C) where channels role alternates between real and imaginary part. The way the squared