9a2903dac097a7f3633f6c828533ed385c7159c9,qucumber/utils/data.py,,extract_refbasis_samples,#,64

Before Change

        if flag == 0:
            tmp.append(train_samples[i])
    z_samples = torch.zeros(len(tmp), num_visible, dtype=torch.double)
    for i in range(len(tmp)):
        for j in range(num_visible):
            z_samples[i][j] = tmp[i][j]
    return z_samples

After Change

    :rtype: torch.Tensor
    
    idx = (
        torch.tensor(train_bases == "Z", dtype=torch.uint8)
        .all(dim=1)
        .to(train_samples.device)
    )

Instances