ops.py 文件源码

def op_sqr_cosine(s_x_, s_y_, axis_=-1, keepdims_=True, eps_=1e-7):
    '''
    squared cosine

    for some occasion, sqrt is not needed
    '''
    s_prod = s_x_ * s_y_
    s_nx = T.sum(T.sqr(s_x_), axis=axis_, keepdims=keepdims_)
    s_ny = T.sum(T.sqr(s_y_), axis=axis_, keepdims=keepdims_)
    return (T.sqr(T.sum(s_prod, axis=axis_, keepdims=keepdims_)) / (s_nx * s_ny + eps_))