4265551c99be9b7dd1d549ecdb13d25bbf98bbf7,geomstats/learning/em_expectation_maximization.py,,gaussianPDF,#Any#Any#Any#Any#Any#Any#,517

Before Change

    // print("sigma",num)
    // print("den ", den)
    // print("pdf max ", (num/den.unsqueeze(0).expand(N, M)).max())
    return num/den.unsqueeze(0).expand(N, M)


def distance(x, y):
    return PoincareDistance2.apply(x, y)

After Change

    // print("den ", den)
    // print("pdf max ", (num/den.unsqueeze(0).expand(N, M)).max())

    den_gs = gs.expand_dims(den,0)
    den_gs = gs.repeat(den_gs, N, axis = 0)

    result = num/den_gs

    return result

Instances