pytorch中.data()与.item(),.numpy(),.detach(),.cuda(),.cpu()的作用

.data:
把变量设置成tensor类型,requires_grad设置为Flase

.item()
把tensor转为python的float类型

.numpy()
将张量转换为与其共享底层存储的 n 维 numpy 数组

.detach()
返回一个与当前 graph 分离的、不再需要梯度的新张量

.cuda()
将张量拷贝到 GPU 上

.cpu()
将张量拷贝到 CPU 上

你可能感兴趣的:(pytorch,pytorch,python)