前言
在機器學習的領域中,我們經常用的資料格式就是 tensor,
但有時為了方便我們查看數值,我們需要將他轉回純量。
範例程式碼
我們能使用 「.item()」 來進行轉換,下面範例程式碼。
import torch
a = torch.tensor([2])
print(a)
print(a.item())
- 結果
在機器學習的領域中,我們經常用的資料格式就是 tensor,
但有時為了方便我們查看數值,我們需要將他轉回純量。
我們能使用 「.item()」 來進行轉換,下面範例程式碼。
import torch
a = torch.tensor([2])
print(a)
print(a.item())