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())
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:pytorch的.item()方法 - Python技术站