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

Question

我得到了一维IntTensor，但我想将其转换为整数。我通过这种方法尝试：

print(dictionary[IntTensor.int()])

但出现错误：

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

谢谢〜

Answer 1

我知道的最简单最干净的方法：

IntTensor.item()

从PyTorch文档：

“将该张量的值作为标准Python数字返回。这仅适用于具有一个元素的张量。其他情况，请参见：meth：~Tensor.tolist“

Answer 2

您可以使用：

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

您使用的密钥是类型的对象autograd.Variable。 给出张量和可用于访问元素的.data索引.0