def op_sqr_c(s_xr_, s_xi_): ''' elemwise complex square ''' return T.sqr(s_xr_) - T.sqr(s_xi_), 2*s_xr_*s_xi_