f097b3e3b0208a23f3b8f90367f32522087025dd,tfdeploy.py,,SparseSegmentSqrtN,#,927

Before Change

    Sparse segmented sum / sqrt(n=len(idxs)) op.
    
    func = lambda _idxs: np.divide(reduce(np.add, a[idxs][_idxs]), np.math.sqrt(len(_idxs)))
    return seg_map(func, a, ids)


@Operation.factory
def Softmax(a):

After Change

    Sparse segmented sum / sqrt(n=len(idxs)) op.
    
    func = lambda _idxs: np.divide(reduce(np.add, a[idxs][_idxs]), np.math.sqrt(len(_idxs)))
    return (seg_map(func, a, ids),)


@Operation.factory
def ArgMin(a, dim):

Instances