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

前言

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

範例程式碼

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

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

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

Licensed under CC BY-NC-SA 4.0