ValueError: can‘t optimize a non-leaf Tensor 数据在cpu 和gpu之间转换的时候涉及到requires_grad和is_leaf的趣事

当我生成一个可导类型的cpu张量的时候,判断其是否为叶子节点,显示为True ,如下面的代码所示:

x = torch.randn(750,2048,requires_grad=True)
print(x.is_leaf)

#输出为:True

但是当我直接把这个生成的cpu类型迁移到gpu上时,有意思的事情发生了:

x = torch.randn(750,2048,requires_grad=True).cuda()
print(x.is_leaf)

#输出为 False

很蒙蔽吧,那么如何让一个生成的张量迁移到gpu上,并且在gpu上也为叶子节点呢?先将数据迁移到gpu上,然后再设置gpu上面的张量的requires_grad为True ,这样就可以了:

x = torch.randn(750,2048).cuda()
x.requires_grad = True
print(x.is_leaf)

#输出结果为: True

 

你可能感兴趣的:(pytorch)