Function tensor2var

GAN/self_attention _gan/utils.py:11–14

(x, grad=False)

def tensor2var(x, grad=False):
    if torch.cuda.is_available():
        x = x.cuda()
    return Variable(x, requires_grad=grad)

def var2tensor(x):
    return x.data.cpu()