分類

展開全部 | 收合全部

分類

展開全部 | 收合全部

【Pytorch】tensor.item() – 從 tensor 轉純量的方法 tensor.item()

前言

在機器學習的領域中,我們經常用的資料格式就是 tensor,
但有時為了方便我們查看數值,我們需要將他轉回純量。

範例程式碼

我們能使用 「.item()」 來進行轉換,下面範例程式碼。

import torch
a = torch.tensor([2])

print(a)
print(a.item())
  • 結果

Howard Weng
Howard Weng

我是 Howard Weng,很多人叫我嗡嗡。這個網站放了我的各種筆記。希望這些筆記也能順便幫助到有需要的人們!如果文章有幫助到你的話,歡迎幫我點讚哦!
另外,因為定位是「個人的隨手筆記」,有些文章內容「⚠️可能我理解有誤⚠️」或「?只寫到一半?」,如果有發現這樣的情況,歡迎在該文章的最下面留言提醒我!我會儘快修正或補上!感謝大家的建議與幫忙,讓網站能變得更好?

文章: 889

★留個言吧!內容有誤或想要補充也歡迎與我討論!