utils.py 文件源码

def generate(dbm, iteration = 1, n = 1):
    even_layer = []
    odd_layer = []
    for i in range(0, dbm.n_odd_layers):
        odd_layer.append(torch.bernoulli((dbm.bias[2*i+1]*0+0.5).view(1,-1).repeat(n, 1)))
    for _ in range(iteration):
        p_even_layer, even_layer = dbm.odd_to_even(odd_layer)
        p_odd_layer, odd_layer = dbm.even_to_odd(even_layer)

    return even_layer[0]