Python PyTorch Tensor.item用法及代碼示例

本文簡要介紹python語言中 torch.Tensor.item 的用法。

用法:

Tensor.item() → number

將此張量的值作為標準 Python 數字返回。這僅適用於具有一個元素的張量。對於其他情況，請參閱 tolist() 。

此操作不可微分。

例子：

>>> x = torch.tensor([1.0])
>>> x.item()
1.0