pytorch中，.item()方法 是得到一個元素張量裏面的元素值
具體就是 用於將一個零維張量轉換成浮點數，比如計算loss，accuracy的值
就比如：
loss = (y_pred - y).pow(2).sum()
print(loss.item())
pytorch中的item()用法
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
pytorch中，.item()方法 是得到一個元素張量裏面的元素值
具體就是 用於將一個零維張量轉換成浮點數，比如計算loss，accuracy的值
就比如：
loss = (y_pred - y).pow(2).sum()
print(loss.item())