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

Posted 2023-02-16

【中文标题】如何在 Pytorch 中将一维 IntTensor 转换为 int

【英文标题】：How to cast a 1-d IntTensor to int in Pytorch

【发布时间】：2018-05-15 06:44:46

【问题描述】：

我得到一个一维 IntTensor，但我想将其转换为整数。 我用这个方法试试：

print(dictionary[IntTensor.int()])

但出现错误：

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

谢谢~

【问题讨论】：

这能回答你的问题吗？ How do I get value of a tensor in PyTorch?

【参考方案1】：

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

IntTensor.item()

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

PyTorch Docs

【参考方案2】：

你可以使用：

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

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