【Pytorch】從 tensor 轉純量的方法 tensor.item() (內附範例程式碼) sample code

➣ Reading Time: 少於 1 minute

前言

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

範例程式碼

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

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

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

預設圖片
Howard Weng
我是 Howard Weng,很多人叫我嗡嗡。這個網站放了我的各種筆記。希望這些筆記也能順便幫助到有需要的人們!如果文章有幫助到你的話,歡迎幫我點讚哦!
文章: 391

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