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

本文简要介绍python语言中 torch.Tensor.item 的用法。

用法:

Tensor.item() → number

将此张量的值作为标准 Python 数字返回。这仅适用于具有一个元素的张量。对于其他情况，请参阅 tolist() 。

此操作不可微分。

例子：

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