364948bf0bb46f9bc3809bbe0a63d09348994444,tests/complexGrads_test.py,,partition,#Any#Any#,26

Before Change

    max_free_energy = free_energies.max()

    f_reduced = free_energies - max_free_energy
    logZ = max_free_energy + f_reduced.exp().sum().log()
    return logZ.exp()


def probability(nn_state, v, Z):
    Evaluates the probability of the given vector(s) of visible

After Change

    // f_reduced = free_energies - max_free_energy
    // logZ = max_free_energy + f_reduced.exp().sum().log()
    // return logZ.exp()
    return torch.tensor(
        nn_state.rbm_am.compute_partition_function(visible_space),
        device=nn_state.device, dtype=torch.double
    )


def probability(nn_state, v, Z):
    Evaluates the probability of the given vector(s) of visible

Instances