loss，loss.cpu().data 及 loss.cpu().detach().numpy()等辨析

原创于 2019-06-04 18:16:00 发布 · 2.2w 阅读

63 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

88 篇文章

订阅专栏

本文详细解析了PyTorch中Tensor数据的操作方法，包括如何将Tensor从GPU转移到CPU，如何从Tensor中提取数据并转换为NumPy数组，以及如何使用.item()方法将0维Tensor转换为Python数字。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

print('1111',loss)
print('2222',loss.data)#tensor且GPU
print('3333',loss.cpu())
print('4444',loss.cpu().data)#tensor且CPU
# print('5555',loss.cpu().data[0])#报错 IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
# print('6666',loss.cpu().numpy())#报错 RuntimeError: Can't call numpy() on Variable that requires grad. Use var.detach().numpy() instead.
print('7777',loss.cpu().detach().numpy())
print('8888',loss.cpu().data.numpy())
print('9999',loss.cpu().item())
print('aaaa',loss.item())#后四者一样，都是把数值取出来