➣ Reading Time: 少於 1 minute

前言

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

範例程式碼

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

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

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