pytorch的.item()方法

python的.item()用于将字典中每对key和value组成一个元组,并把这些元组放在列表中返回
例如
person={‘name’:‘lizhong’,‘age’:‘26’,‘city’:‘BeiJing’,‘blog’:‘www.jb51.net’}

for key,value in person.items():
print ‘key=’,key,’,value=’,value

而pytorch中的.item()用于将一个零维张量转换成浮点数,比如loss = (y_pred - y).pow(2).sum()

print(loss.item())

原文地址:https://www.cnblogs.com/tingtin/p/13544941.html