ops.py 文件源码

def op_norm2_c(s_xr_, s_xi_, axis_=-1, use_mean_=False, keepdims_=True):
    '''
    Complex squared L2 norm
    '''
    op_sum = T.sum if not use_mean_ else T.mean
    return op_sum(T.sqr(s_xr_) + T.sqr(s_xi_), axis=axis_, keepdims=keepdims_)