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())


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM