python - 如何在 PyTorch 中获取张量的值？

Question

打印张量给出：

>>> x = torch.tensor([3])
>>> print(x)
tensor([3])

同样索引其.data给出：

>>> x.data[0]
tensor(3)

我如何获得价值3？

标签： pythonpytorchtensor

Answer 1

您可以使用x.item()从具有一个元素的张量中获取 Python 数字。