torch.Tensor.item()

张量.item() 数字

返回此张量的值,作为标准的Python数字。这仅适用于包含单个元素的张量。对于其他情况,请参见tolist()

此操作不具备可微性。

示例:

>>> x = torch.tensor([1.0])
>>> x.item()
1.0
本页目录