如何在Pytorch中将1维IntTensor转换为int

Question

41

如何将1-D IntTensor转换为整数？这样做：

IntTensor.int()

引发错误：

KeyError: Variable containing:
 423
[torch.IntTensor of size 1]

- Ruben

1

你能否更新被接受的答案？ - Metehan

2个回答

-3

您可以使用：

print(dictionary[IntTensor.data[0]])

您正在使用的密钥是一个autograd.Variable对象。 .data提供张量，索引0可用于访问元素。

- skb

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Marcin · Accepted Answer

我知道的最简单、最清晰的方法：

IntTensor.item()

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